Skip to content

Commit ae2668e

Browse files
changed default train config
1 parent ef62758 commit ae2668e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/pytorch/FastCells/training_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def __init__(self):
3636
class OptimizerOptions:
3737
def __init__(self):
3838
self.weight_decay = 1e-5
39-
self.momentum = 0 # RMSprop
39+
self.momentum = 0.9 # RMSprop
4040
self.centered = False # RMSprop
4141
self.alpha = 0 # ASGD, RMSprop
4242
self.eps = 1e-8

0 commit comments

Comments
 (0)