Removes unused weighting parameters
Browse filesRemoves unused `score_weight` and `class_weight` parameters from the `train_pointnet` function call.
These parameters are no longer used within the function, simplifying the call and improving code clarity.
train_pnet_class_cluster.py
CHANGED
|
@@ -10,4 +10,4 @@ if __name__ == "__main__":
|
|
| 10 |
os.makedirs(model_save_path, exist_ok=True)
|
| 11 |
|
| 12 |
# Train the model
|
| 13 |
-
train_pointnet(dataset_path, model_save_path, epochs=100, batch_size=128, lr=0.001
|
|
|
|
| 10 |
os.makedirs(model_save_path, exist_ok=True)
|
| 11 |
|
| 12 |
# Train the model
|
| 13 |
+
train_pointnet(dataset_path, model_save_path, epochs=100, batch_size=128, lr=0.001)
|