If you’re not passing the model then you need to set the model_init (take a look here 2) You can basically pass the configuration of your model (if you don’t want to start from a checkpoint) and Trainer will instantiate the model when the train function is called.