diff --git a/train_classification_geoerror.py b/train_classification_geoerror.py
index e87f5b0..ad00da1 100644
--- 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()