From 1b997f7bb96abff473e169634245b2c39b3f78d7 Mon Sep 17 00:00:00 2001
From: li-xl <1905692338@qq.com>
Date: Thu, 20 Oct 2022 20:19:11 +0800
Subject: [PATCH] change readme

---
 README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index ff31695..710b22c 100644
--- a/README.md
+++ b/README.md
@@ -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),