From 0a62df22aa1bc802f5b050de6eba87a3e298d883 Mon Sep 17 00:00:00 2001 From: Yishuo Wang Date: Wed, 21 Dec 2022 16:31:15 +0800 Subject: [PATCH] Nano: quick fix typo (#7064) --- .../readthedocs/source/doc/Nano/Overview/pytorch_train.md | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/readthedocs/source/doc/Nano/Overview/pytorch_train.md b/docs/readthedocs/source/doc/Nano/Overview/pytorch_train.md index 984834f0..5d2d0156 100644 --- a/docs/readthedocs/source/doc/Nano/Overview/pytorch_train.md +++ b/docs/readthedocs/source/doc/Nano/Overview/pytorch_train.md @@ -35,7 +35,7 @@ For example, from bigdl.nano.pytorch import Trainer lightning_module = Trainer.compile(pytorch_module, loss, optimizer) -trainer = Trainer(max_epoch=10) +trainer = Trainer(max_epochs=10) trainer.fit(lightning_module, train_loader) ``` @@ -69,7 +69,7 @@ class MyModule(LightningModule): # LightningModule definition lightning_module = MyModule() -trainer = Trainer(max_epoch=10) +trainer = Trainer(max_epochs=10) trainer.fit(lightning_module, train_loader) ``` @@ -90,7 +90,7 @@ BigDL-Nano integrates IPEX in `Trainer` and `TorchNano`. Users can turn on IPEX from bigdl.nano.pytorch import Trainer - trainer = Trainer(max_epoch=10, use_ipex=True) + trainer = Trainer(max_epochs=10, use_ipex=True) trainer.fit(...) .. tab:: TorchNano @@ -123,7 +123,7 @@ You can just set the `num_processes` parameter in the `Trainer` or `TorchNano` c from bigdl.nano.pytorch import Trainer - trainer = Trainer(max_epoch=10, num_processes=4) + trainer = Trainer(max_epochs=10, num_processes=4) trainer.fit(...) .. tab:: TorchNano