Update FFV1MT_MS.py

FFV1MT_MS.py

@@ -218,7 +218,8 @@ class FFV1DNN(nn.Module):
             results_dict["flow_seq"] = flows_up
             results_dict["flow_seq_bi"] = flows_bi
         return results_dict
-
+        
+    @torch.no_grad()
     def forward_viz(self, image_list, layer=None, x=50, y=50):
         x = x / 100
         y = y / 100