Changed some code haaa

notebook_helpers.py

@@ -5,9 +5,6 @@ from utils.utils_func import convert_dict
 from utils.surfaces import Surface
 import numpy as np
 
-
-device = "cuda:0"
-
 def load_data(file, cache_path, name, num_evecs=128, make_cache=False, factor=None):
     if factor is None:
         verts_shape, faces, vnormals, area_shape, center_shape = load_mesh(file, return_vnormals=True)
@@ -23,9 +20,8 @@ def load_data(file, cache_path, name, num_evecs=128, make_cache=False, factor=No
     data_dict['name'] = name
     data_dict['normals'] = vnormals
     data_dict['vertices'] = verts_shape
-    data_dict_torch = convert_dict(data_dict, device)
     #batchify_dict(data_dict_torch)
-    return data_dict_torch, area_shape
+    return data_dict, area_shape
 
 def get_map_info(file_1, file_2, dict_1, dict_2, dataset):
     shape_dict, target_dict = dict_1, dict_2