change to cpu

model_utils.py

@@ -42,7 +42,7 @@ def load_model(config: str, weight: str, model_def, device):
 
     opt = parse_configs(config)
     model = model_def(opt)
-    cp = torch.load(weight)
+    cp = torch.load(weight, map_location=device)
 
     models = model.get_models()
     for k, m in models.items():