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