sindhuhegde commited on
Commit
129cbf8
·
1 Parent(s): 8161a68

Update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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