assign map_location

leo/inference.py

@@ -92,7 +92,7 @@ def load_data(scan_id):
     one_scan['obj_box'] = obj_box_size
     # load point feat
     feat_pth = os.path.join(ASSET_DIR, f'inputs/{scan_id}', 'obj_feats.pth')
-    one_scan['obj_feats'] = torch.load(feat_pth).to('cpu')
+    one_scan['obj_feats'] = torch.load(feat_pth, map_location='cpu')
     #  convert to pq3d input
     obj_labels = one_scan['inst_labels'] # N
     obj_pcds = one_scan['obj_pcds']