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

Automatically unwrap attributes access of _LiteModule (#5392)

Browse source 
* automatically unwrap attributes access of _LiteModule and add test for it
* delete redundant documents
This commit is contained in:
Yishuo Wang 2022-08-18 10:57:52 +08:00 committed by GitHub
parent df75c72d91
commit e62b8d1fc5
1 changed files with 0 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
docs/readthedocs/source/doc/Nano/QuickStart/pytorch_nano.md
View file

@ -111,7 +111,6 @@ We only need the following steps:
- copy all lines of code into the `train` method of `MyNano` - copy all lines of code into the `train` method of `MyNano`
- add one line to setup model, optimizer and dataloader - add one line to setup model, optimizer and dataloader
- replace the `loss.backward()` with `self.backward(loss)` - replace the `loss.backward()` with `self.backward(loss)`
- (**optional, only required when you use custom fields or methods of the model**) after calling `self.setup`, the returned model is a wrapper module around the original one which only has some common methods, such as `train`, `forward`, etc. if you need to access other fields or methods of the original model, you can replace the `model.xxxx` with `model.module.xxxx`
```python ```python
import os import os