Increases batch size for training

Increases the batch size from 128 to 512 during training.

This change aims to improve training efficiency and potentially lead to faster convergence by processing more data in each iteration.

train_pnet_class_cluster.py

@@ -10,4 +10,4 @@ if __name__ == "__main__":
     os.makedirs(model_save_path, exist_ok=True)
     
     # Train the model
-    train_pointnet(dataset_path, model_save_path, epochs=100, batch_size=128, lr=0.001)
+    train_pointnet(dataset_path, model_save_path, epochs=100, batch_size=512, lr=0.001)