Update FFV1MT_MS.py

FFV1MT_MS.py

@@ -260,9 +260,21 @@ class FFV1DNN(nn.Module):
         w = int(MT_size[1] * x)
         attention = plt_attention(attn_viz, h=h, w=w)
         print("done")
+        plt.clf()
+        plt.cla()
+        plt.close()
         results_dict["activation"] = activation
+        plt.clf()
+        plt.cla()
+        plt.close()
         results_dict["attention"] = attention
+        plt.clf()
+        plt.cla()
+        plt.close()
         results_dict["flow"] = flow
+        plt.clf()
+        plt.cla()
+        plt.close()
 
         return results_dict