default warm up

This commit is contained in:
Varuna Jayasiri
2020-12-23 13:57:34 +05:30
parent 634809af3c
commit b05faa9b98

View File

@ -30,7 +30,7 @@ class OptimizerConfigs(BaseConfigs):
momentum: float = 0.5 momentum: float = 0.5
amsgrad: bool = False amsgrad: bool = False
warmup: int = 0 warmup: int = 2_000
degenerated_to_sgd: bool = True degenerated_to_sgd: bool = True
rectify: bool = True rectify: bool = True
d_model: int d_model: int