add some update

wyddmw · May 5, 2021 · 57621cd · 57621cd
1 parent 8746c60
commit 57621cd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train_classification_geoerror.py b/train_classification_geoerror.py
@@ -90,6 +90,7 @@
 for epoch in range(opt.nepoch):
     scheduler.step()
     for i, (point, label) in enumerate(dataloader, 0):
+        points = points.transpose(2, 1)
         points = point.cuda()
         target = label.cuda()