memcpy is super slow in cuda kernels....

This commit is contained in:
erikwijmans
2018-01-29 18:22:27 -05:00
parent 009c30e5e3
commit c4ddd6b099
7 changed files with 30 additions and 14 deletions
+3 -1
View File
@@ -8,7 +8,8 @@ sys.path.append(BASE_DIR)
def _get_data_files(list_filename):
return [line.rstrip()[5:] for line in open(list_filename)]
with open(list_filename) as f:
return [line.rstrip()[5:] for line in f]
def _load_data_file(name):
@@ -80,6 +81,7 @@ class ModelNet40Cls(data.Dataset):
def set_num_points(self, pts):
self.num_points = pts
self.actual_number_of_points = pts
def randomize(self):
self.actual_number_of_points = min(