use sgdr scheduler tf impl

mlr-org / mlr3keras

Deep learning for mlr3

GNU Lesser General Public License v3.0

36 stars 3 forks source link

Closed pfistfl closed 4 years ago

pfistfl commented 4 years ago

callback_learning_rate_scheduler(tf$keras$experimental$CosineDecayRestarts(0.001, 100, t_mul = 2, m_mul = .8))

pfistfl commented 4 years ago

Should be fixed