Change map_location to be 'cpu' (#1461)

* Change map_location to be 'cpu'

If you are on a CPU-only machine, it will give an error otherwise. Model averaging should not require a GPU; moreover, it may be faster to use CPU rather than move all models to the GPU to average them.

Author: ymoslem

tools/average_models.py

@@ -10,7 +10,7 @@ def average_models(model_files):
     avg_generator = None
 
     for i, model_file in enumerate(model_files):
-        m = torch.load(model_file)
+        m = torch.load(model_file, map_location='cpu')
         model_weights = m['model']
         generator_weights = m['generator']