Disable grad in eval.

facebookresearch · Oct 2, 2019 · ab98883 · ab98883
1 parent 237f6ff
commit ab98883
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -265,7 +265,8 @@ def evaluate_one_epoch():
 
         # Forward pass
         inputs = {'point_clouds': batch_data_label['point_clouds']}
-        end_points = net(inputs)
+        with torch.no_grad():
+            end_points = net(inputs)
 
         # Compute loss
         for key in batch_data_label: