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. We test the speed on the following model and choose 10 scenes from ScanNet as the dataset.
```python ```python
import jsparse.nn as spnn
from jittor import nn
algorithm = "cuda" algorithm = "cuda"
# you can turn it to "jittor"
model = nn.Sequential( model = nn.Sequential(
spnn.Conv3d(3, 32, 2), spnn.Conv3d(3, 32, 2),
spnn.BatchNorm(32), spnn.BatchNorm(32),