Fix torchhub due to numerical changes in torch.sum (#2361)

Author: fmassa

test/test_hub.py

@@ -13,7 +13,7 @@ def sum_of_model_parameters(model):
     return s
 
 
-SUM_OF_PRETRAINED_RESNET18_PARAMS = -12703.99609375
+SUM_OF_PRETRAINED_RESNET18_PARAMS = -12703.9931640625
 
 
 @unittest.skipIf('torchvision' in sys.modules,
@@ -31,8 +31,9 @@ def test_load_from_github(self):
             'resnet18',
             pretrained=True,
             progress=False)
-        self.assertEqual(sum_of_model_parameters(hub_model).item(),
-                         SUM_OF_PRETRAINED_RESNET18_PARAMS)
+        self.assertAlmostEqual(sum_of_model_parameters(hub_model).item(),
+                               SUM_OF_PRETRAINED_RESNET18_PARAMS,
+                               places=2)
 
     def test_set_dir(self):
         temp_dir = tempfile.gettempdir()
@@ -42,8 +43,9 @@ def test_set_dir(self):
             'resnet18',
             pretrained=True,
             progress=False)
-        self.assertEqual(sum_of_model_parameters(hub_model).item(),
-                         SUM_OF_PRETRAINED_RESNET18_PARAMS)
+        self.assertAlmostEqual(sum_of_model_parameters(hub_model).item(),
+                               SUM_OF_PRETRAINED_RESNET18_PARAMS,
+                               places=2)
         self.assertTrue(os.path.exists(temp_dir + '/pytorch_vision_master'))
         shutil.rmtree(temp_dir + '/pytorch_vision_master')