mlr3torch
mlr3torch copied to clipboard
Callbacks
Some predefined callbacks we might want to support:
- [ ] LR scheduler
- [ ] Writing files in tensorboard format: https://github.com/mlverse/tfevents
- [ ] Experiment Tracking using guildai-r: https://github.com/guildai/guildai-r
- [x] Model Saver
- [x] History / Logger: maybe only to this every N epochs, maybe depends on giving callbacks parameters / construction arguments