[fbsync] Fix minor issues with model tests. (#5576)

Reviewed By: vmoens

Differential Revision: D34878988

fbshipit-source-id: ec196a6ed81f81caa0624749c0420b7419c1d8f6

Author: datumbox

test/test_models.py

@@ -81,7 +81,7 @@ def _get_expected_file(name=None):
 
     # Note: for legacy reasons, the reference file names all had "ModelTest.test_" in their names
     # We hardcode it here to avoid having to re-generate the reference files
-    expected_file = expected_file = os.path.join(expected_file_base, "ModelTester.test_" + name)
+    expected_file = os.path.join(expected_file_base, "ModelTester.test_" + name)
     expected_file += "_expect.pkl"
 
     if not ACCEPT and not os.path.exists(expected_file):
@@ -665,6 +665,7 @@ def check_out(out):
         assert len(out) == 1
 
         def compact(tensor):
+            tensor = tensor.cpu()
             size = tensor.size()
             elements_per_sample = functools.reduce(operator.mul, size[1:], 1)
             if elements_per_sample > 30: