Learning rate scheduling with Flax does not work
Ouverte
feature request
- Langage dominant
- Python
- Étoiles
- 572
- Forks
- 55
- Métriques de merge des PR
- Aucune PR mergée en 30 j
Description
### 🚀 The feature, motivation and pitch
The torch trainer has LR scheduling. The flax trainer should as well.
### Alternatives
* None
### Additional context
The trainer takes an `lr_scheduler` argument, but ignores it.
Guide de contribution
Ouvrir le guide de contribution
Évaluation
Cette issue n'a pas encore été évaluée.