diff --git a/test_davis_videos.py b/test_davis_videos.py
index e02db64..49a839f 100644
--- a/test_davis_videos.py
+++ b/test_davis_videos.py
@@ -50,4 +50,5 @@
     print(i)
     stacked_img = data[0]
     targets = data[1]
-    model.run_and_save_DAVIS(stacked_img, targets, save_path)
+    with torch.no_grad():
+      model.run_and_save_DAVIS(stacked_img, targets, save_path)