ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix: update pytorch estimator backend. (#6721)

Browse source
This commit is contained in:
Cengguang Zhang 2022-11-22 16:50:33 +08:00 committed by GitHub
parent e249e32c6f
commit 7f2beefb7b
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-quickstart.md
View file

@ -116,7 +116,7 @@ First, Create an Estimator
from bigdl.orca.learn.pytorch import Estimator
from bigdl.orca.learn.metrics import Accuracy
est = Estimator.from_torch(model=model, optimizer=adam, loss=criterion, metrics=[Accuracy()])
est = Estimator.from_torch(model=model, optimizer=adam, loss=criterion, metrics=[Accuracy()], backend="bigdl")
```
Next, fit and evaluate using the Estimator