Personality Captions interactive mode fix (facebookresearch#2963)

* Add fix * Fixes
analysmith · Aug 12, 2020 · 2345a05 · 2345a05
1 parent 25f66a7
commit 2345a05
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/projects/personality_captions/interactive.py b/projects/personality_captions/interactive.py
@@ -275,8 +275,6 @@ def setup_interactive():
         )
     opt['task'] = 'parlai.agents.local_human.local_human:LocalHumanAgent'
     opt['image_mode'] = 'resnet152'
-    opt['no_cuda'] = True
-    opt['override']['no_cuda'] = True
     SHARED['opt'] = opt
     SHARED['image_loader'] = ImageLoader(opt)
 

diff --git a/projects/personality_captions/transresnet/modules.py b/projects/personality_captions/transresnet/modules.py
@@ -399,9 +399,9 @@ def choose_best_caption(
         for img_index in range(len(context_encoded)):
             context_encoding = context_encoded[img_index : img_index + 1, :]
             scores = torch.mm(
-                candidates_encoded[img_index]
+                candidates_encoded[img_index].to(context_encoding)
                 if not one_cand_set
-                else candidates_encoded,
+                else candidates_encoded.to(context_encoding),
                 context_encoding.transpose(0, 1),
             )
             if k >= 1: