Skip to content

Commit e48e52e

Browse files
committed
Add generic repr to Optimizers.
1 parent 95c45b4 commit e48e52e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

DeepFried2/optimizers/Optimizer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,3 +24,6 @@ def update_parameters(self, model):
2424

2525
def get_updates(self, params, grads):
2626
raise NotImplementedError
27+
28+
def __repr__(self):
29+
return type(self).__name__ + "(" + ", ".join(k+"="+str(v) for k,v in self.hyperparams.items()) + ")"

0 commit comments

Comments
 (0)