discard GPU usage

src/pixrender/learning/utils.py

@@ -92,7 +92,7 @@ def save_model(model, points, normals, colors, output_path):
 
 
 def load_model(path):
-    model_dic = torch.load(path)
+    model_dic = torch.load(path, map_location=torch.device("cpu"))
     model_state_dict = model_dic["model"]
     points = model_dic["point_cloud"]
     normals = model_dic["normals"]