fix map location error. all previous commits were made by me, config was wrong for git

run.py

@@ -155,7 +155,7 @@ kp_detector = KPDetector(**config['model_params']['kp_detector_params'], **confi
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 cpu = False if torch.cuda.is_available() else True
     
-g_checkpoint = torch.load("checkpoints/generator.pth", map_location = device)
+g_checkpoint = torch.load("checkpoints/generator.pth", map_location=device)
 kp_checkpoint = torch.load("checkpoints/kp_detector.pth", map_location=device)
 
 ckp_generator = OrderedDict((k.replace('module.',''),v) for k,v in g_checkpoint.items())
@@ -165,8 +165,8 @@ kp_detector.load_state_dict(ckp_kp_detector)
 
 depth_encoder = depth.ResnetEncoder(18, False)
 depth_decoder = depth.DepthDecoder(num_ch_enc=depth_encoder.num_ch_enc, scales=range(4))
-loaded_dict_enc = torch.load('checkpoints/encoder.pth')
-loaded_dict_dec = torch.load('checkpoints/depth.pth')
+loaded_dict_enc = torch.load('checkpoints/encoder.pth', map_location=device)
+loaded_dict_dec = torch.load('checkpoints/depth.pth', map_location=device)
 
 filtered_dict_enc = {k: v for k, v in loaded_dict_enc.items() if k in depth_encoder.state_dict()}
 depth_encoder.load_state_dict(filtered_dict_enc)