lorocksUMD commited on
Commit
eb1817f
·
verified ·
1 Parent(s): 05f94d6

Add a debug print to app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -103,6 +103,9 @@ if __name__ == "__main__":
103
 
104
  with torch.no_grad():
105
  # audio_feats = model.forward_audio({"audio": audio.cuda()})
 
 
 
106
  audio_feats = model.forward_audio({"audio": audio})
107
  audio_feats = {k: v.cpu() for k, v in audio_feats.items()}
108
  # image_feats = model.forward_image({"frames": frames.unsqueeze(0).cuda()}, max_batch_size=2)
 
103
 
104
  with torch.no_grad():
105
  # audio_feats = model.forward_audio({"audio": audio.cuda()})
106
+ print("---"*10)
107
+ print(model.full_train)
108
+ print("---"*10)
109
  audio_feats = model.forward_audio({"audio": audio})
110
  audio_feats = {k: v.cpu() for k, v in audio_feats.items()}
111
  # image_feats = model.forward_image({"frames": frames.unsqueeze(0).cuda()}, max_batch_size=2)