sunana commited on
Commit
7e068f4
1 Parent(s): 24aba9b

Update FFV1MT_MS.py

Browse files
Files changed (1) hide show
  1. FFV1MT_MS.py +0 -9
FFV1MT_MS.py CHANGED
@@ -260,17 +260,8 @@ class FFV1DNN(nn.Module):
260
  w = int(MT_size[1] * x)
261
  attention = plt_attention(attn_viz, h=h, w=w)
262
  print("done")
263
- plt.clf()
264
- plt.cla()
265
- plt.close()
266
  results_dict["activation"] = activation
267
- plt.clf()
268
- plt.cla()
269
- plt.close()
270
  results_dict["attention"] = attention
271
- plt.clf()
272
- plt.cla()
273
- plt.close()
274
  results_dict["flow"] = flow
275
  plt.clf()
276
  plt.cla()
 
260
  w = int(MT_size[1] * x)
261
  attention = plt_attention(attn_viz, h=h, w=w)
262
  print("done")
 
 
 
263
  results_dict["activation"] = activation
 
 
 
264
  results_dict["attention"] = attention
 
 
 
265
  results_dict["flow"] = flow
266
  plt.clf()
267
  plt.cla()