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 |
|