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

Update pytorch_train.md (#4786)

Browse source 
* Update pytorch_train.md

Add description about torchvision version

* Update pytorch_train.md
This commit is contained in:
Mingzhi Hu 2022-06-09 08:39:44 +08:00 committed by GitHub
parent 20ad708010
commit 2c4623f8da
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
docs/readthedocs/source/doc/Nano/QuickStart/pytorch_train.md
View file

@ -51,6 +51,7 @@ Note: BigDL-Nano does not install IPEX by default. You can install IPEX using th
```bash ```bash
python -m pip install torch_ipex==1.9.0 -f https://software.intel.com/ipex-whl-stable python -m pip install torch_ipex==1.9.0 -f https://software.intel.com/ipex-whl-stable
python -m pip install torchvision==0.10.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
``` ```
#### Multi-instance Training #### Multi-instance Training
@ -86,4 +87,4 @@ data_transform = transforms.Compose([
train_set = ImageFolder(train_path, data_transform) train_set = ImageFolder(train_path, data_transform)
train_loader = DataLoader(train_set, batch_size=batch_size, shuffle=True) train_loader = DataLoader(train_set, batch_size=batch_size, shuffle=True)
trainer.fit(module, train_loader) trainer.fit(module, train_loader)
``` ```