Bugbash for speedup example. (#3996)

examples/model_compress/pruning/speedup/speedup_nanodet.py

@@ -12,7 +12,7 @@ NanoDet model can be installed from https://github.com/RangiLyu/nanodet.git
 cfg_path = r"nanodet/config/nanodet-RepVGG-A0_416.yml"
 load_config(cfg, cfg_path)
 
-model = build_model(cfg.model)
+model = build_model(cfg.model).cpu()
 dummy_input  = torch.rand(8, 3, 416, 416)
 
 op_names = []

examples/model_compress/pruning/speedup/speedup_yolov3.py

@@ -10,7 +10,7 @@ prefix = '/home/user/PyTorch-YOLOv3' # replace this path with yours
 # Load the YOLO model
 model = models.load_model(
   "%s/config/yolov3.cfg" % prefix, 
-  "%s/yolov3.weights" % prefix)
+  "%s/yolov3.weights" % prefix).cpu()
 model.eval()
 dummy_input = torch.rand(8, 3, 320, 320)
 model(dummy_input)