* [ ] register all hooks that are available * [ ] implement LR scheduler * [ ] implement Model Saver * [ ] Print validation loss in torch.progress