pytorch-lightning如何设置训练epoch

Trainer初始化时添加max_epochs参数

# init model
autoencoder = LitAutoEncoder()

trainer = pl.Trainer(gpus=8 , max_epochs=50)
trainer.fit(autoencoder, train_loader)

你可能感兴趣的:(pytorch,python,深度学习)