Spaces:
Runtime error
Runtime error
Commit
•
f827fdd
1
Parent(s):
6be2b30
Update app.py
Browse files
app.py
CHANGED
|
@@ -85,7 +85,8 @@ def forward(model, save_dir, query):
|
|
| 85 |
src_txt = src_txt.cuda(args.gpu_id)
|
| 86 |
src_vid_mask = src_vid_mask.cuda(args.gpu_id)
|
| 87 |
src_txt_mask = src_txt_mask.cuda(args.gpu_id)
|
| 88 |
-
|
|
|
|
| 89 |
with torch.no_grad():
|
| 90 |
output = model(src_vid=src_vid, src_txt=src_txt, src_vid_mask=src_vid_mask, src_txt_mask=src_txt_mask)
|
| 91 |
|
|
|
|
| 85 |
src_txt = src_txt.cuda(args.gpu_id)
|
| 86 |
src_vid_mask = src_vid_mask.cuda(args.gpu_id)
|
| 87 |
src_txt_mask = src_txt_mask.cuda(args.gpu_id)
|
| 88 |
+
|
| 89 |
+
model.eval()
|
| 90 |
with torch.no_grad():
|
| 91 |
output = model(src_vid=src_vid, src_txt=src_txt, src_vid_mask=src_vid_mask, src_txt_mask=src_txt_mask)
|
| 92 |
|