Fixes in evaluation_basic.rst (#18843)

2023-10-27 05:36:30 -05:00 · 2023-10-27 05:36:30 -05:00 · af44205541
parent 4d030c4872
commit af44205541
1 changed files with 2 additions and 1 deletions
--- 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)