Disable grad in eval.

facebookresearch · Oct 2, 2019 · 886f83d · 886f83d
1 parent 69320e1
commit 886f83d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/demo.py b/demo.py
@@ -88,7 +88,8 @@ def preprocess_point_cloud(point_cloud):
     # Model inference
     inputs = {'point_clouds': torch.from_numpy(pc).to(device)}
     tic = time.time()
-    end_points = net(inputs)
+    with torch.no_grad():
+        end_points = net(inputs)
     toc = time.time()
     print('Inference time: %f'%(toc-tic))
     end_points['point_clouds'] = inputs['point_clouds']