Trainer
Trainer copied to clipboard
made a change so that model can be saved when starting from checkpouint