trainer.py

trainer = Trainer(
  callbacks=callbacks,
  accelerator= "auto",
  auto_scale_batch_size= False,
  check_val_every_n_epoch= 1,
  devices= 1,
  gpus= None,
  max_epochs= 1,
  num_sanity_val_steps= 0,
  val_check_interval= 1.0,
)
trainer.fit(model=model, datamodule=datamodule)