Trainer初始化时添加max_epochs参数
# init model autoencoder = LitAutoEncoder() trainer = pl.Trainer(gpus=8 , max_epochs=50) trainer.fit(autoencoder, train_loader)