jittor
/
JSparse
7
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

change readme

This commit is contained in:
li-xl 2022-10-20 20:19:11 +08:00
parent 3df4cc83a0
commit 1b997f7bb9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@ -79,8 +79,9 @@ Because the Jittor framework is fast, inference and training are faster than PyT
We test the speed on the following model and choose 10 scenes from ScanNet as the dataset.
```python
import jsparse.nn as spnn
from jittor import nn
algorithm = "cuda"
# you can turn it to "jittor"
model = nn.Sequential(
spnn.Conv3d(3, 32, 2),
spnn.BatchNorm(32),