Spaces:
Running on Zero
Running on Zero
Commit ·
129cbf8
1
Parent(s): 8161a68
Update app
Browse files
app.py
CHANGED
|
@@ -1100,12 +1100,12 @@ def get_embeddings(video_sequences, audio_sequences, model, calc_aud_emb=True):
|
|
| 1100 |
vid_emb = model.forward_vid(video_inp.to(device), return_feats=False)
|
| 1101 |
vid_emb = torch.mean(vid_emb, axis=-1)
|
| 1102 |
|
| 1103 |
-
video_emb.append(vid_emb.cpu())
|
| 1104 |
|
| 1105 |
if calc_aud_emb:
|
| 1106 |
audio_inp = audio_sequences[i:i+batch_size, ]
|
| 1107 |
aud_emb = model.forward_aud(audio_inp.to(device))
|
| 1108 |
-
audio_emb.append(aud_emb.cpu())
|
| 1109 |
|
| 1110 |
# torch.cuda.empty_cache()
|
| 1111 |
|
|
|
|
| 1100 |
vid_emb = model.forward_vid(video_inp.to(device), return_feats=False)
|
| 1101 |
vid_emb = torch.mean(vid_emb, axis=-1)
|
| 1102 |
|
| 1103 |
+
video_emb.append(vid_emb.detach().cpu())
|
| 1104 |
|
| 1105 |
if calc_aud_emb:
|
| 1106 |
audio_inp = audio_sequences[i:i+batch_size, ]
|
| 1107 |
aud_emb = model.forward_aud(audio_inp.to(device))
|
| 1108 |
+
audio_emb.append(aud_emb.detach().cpu())
|
| 1109 |
|
| 1110 |
# torch.cuda.empty_cache()
|
| 1111 |
|