Spaces:
No application file
No application file
Add a debug print to app.py
Browse files
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)
|