AlbertoCenzato/pytorch_ext

Provide an easy way to instantiate common ModelTrainer-callbacks couplings

AlbertoCenzato opened this issue · 2 comments

AlbertoCenzato commented
Provide an easy way to instantiate common ModelTrainer-callbacks couplings
AlbertoCenzato commented

Added a function, create_model_trainer(), that instantiates a ModelTrainer and adds TrainingTimeEstimation and BatchStatistics. Model checkpointing is still to be added

AlbertoCenzato commented

Added Checkpoint to the callbacks added to ModelTrainer by create_model_trainer()