Learning rate scheduling with Flax does not work
Abierto
feature request
- Lenguaje dominante
- Python
- Estrellas
- 572
- Forks
- 55
- Métricas de merge de PR
- Sin PR fusionados en 30 d
Descripción
### 🚀 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.
Guía de contribución
Evaluación
Este issue todavía no se ha evaluado.