fix torch_nano document link error and small change (#6257)

This commit is contained in:
Yishuo Wang 2022-10-24 14:04:02 +08:00 committed by GitHub
parent f54b9b1915
commit 6a8cdd71de
3 changed files with 6 additions and 8 deletions

View file

@ -74,8 +74,6 @@ class MyNano(TorchNano) :
MyNano().train(...) MyNano().train(...)
``` ```
- note: see [this tutorial](./pytorch_nano.html) for details about our `TorchNano`.
Our `TorchNano` also integrates IPEX and distributed training optimizations. For example, Our `TorchNano` also integrates IPEX and distributed training optimizations. For example,
```python ```python

View file

@ -11,7 +11,7 @@
> ![](../../../../image/GitHub-Mark-32px.png)[View source on GitHub][Nano_pytorch_nano] > ![](../../../../image/GitHub-Mark-32px.png)[View source on GitHub][Nano_pytorch_nano]
In this guide we'll describe how to use BigDL-Nano to accelerate custom training loop easily with very few changes In this guide we will describe how to use BigDL-Nano to accelerate custom training loop easily with very few changes
--------------------------- ---------------------------

View file

@ -2,7 +2,7 @@
**In this guide we'll demonstrate how to use BigDL-Nano to accelerate custom train loop easily with very few changes.** **In this guide we'll demonstrate how to use BigDL-Nano to accelerate custom train loop easily with very few changes.**
### **Step 0: Prepare Environment** ### Step 0: Prepare Environment
We recommend using [conda](https://docs.conda.io/projects/conda/en/latest/user-guide/install/) to prepare the environment. Please refer to the [install guide](../../UserGuide/python.md) for more details. We recommend using [conda](https://docs.conda.io/projects/conda/en/latest/user-guide/install/) to prepare the environment. Please refer to the [install guide](../../UserGuide/python.md) for more details.
@ -15,7 +15,7 @@ pip install --pre --upgrade bigdl-nano[pytorch]
source bigdl-nano-init source bigdl-nano-init
``` ```
### **Step 1: Load the Data** ### Step 1: Load the Data
Import Cifar10 dataset from torch_vision and modify the train transform. You could access [CIFAR10](https://www.cs.toronto.edu/~kriz/cifar.html) for a view of the whole dataset. Import Cifar10 dataset from torch_vision and modify the train transform. You could access [CIFAR10](https://www.cs.toronto.edu/~kriz/cifar.html) for a view of the whole dataset.
@ -49,7 +49,7 @@ def create_dataloader(data_path, batch_size):
return train_loader return train_loader
``` ```
### **Step 2: Define the Model** ### Step 2: Define the Model
You may define your model in the same way as the standard PyTorch models. You may define your model in the same way as the standard PyTorch models.
@ -70,7 +70,7 @@ class ResNet18(nn.Module):
return self.model(x) return self.model(x)
``` ```
### **Step 3: Define Train Loop** ### Step 3: Define Train Loop
Suppose the custom train loop is as follows: Suppose the custom train loop is as follows:
@ -149,7 +149,7 @@ class MyNano(TorchNano):
print(f'avg_loss: {total_loss / num}') print(f'avg_loss: {total_loss / num}')
``` ```
### **Step 4: Run with Nano TorchNano** ### Step 4: Run with Nano TorchNano
```python ```python
MyNano().train() MyNano().train()