diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 9883b51..ed6a783 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -285,7 +285,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... OptimizerClass = nn.RMSprop if adabelief: lr_dropout = 1.0 - lr_cos = 2000 + #lr_cos = 2000 OptimizerClass = nn.AdaBelief clipnorm = 1.0 if self.options['clipgrad'] else 0.0