diff --git a/docs/source-pytorch/common/evaluation_basic.rst b/docs/source-pytorch/common/evaluation_basic.rst index 80da873a45..823f1aba74 100644 --- a/docs/source-pytorch/common/evaluation_basic.rst +++ b/docs/source-pytorch/common/evaluation_basic.rst @@ -124,7 +124,8 @@ To run the validation loop, pass in the validation set to **.fit** train_loader = DataLoader(train_set) valid_loader = DataLoader(valid_set) + model = LitAutoEncoder(...) # train with both splits - trainer = Trainer() + trainer = pl.Trainer() trainer.fit(model, train_loader, valid_loader)