jskvrna commited on
Commit
296e145
·
1 Parent(s): 79c53b0

Removes unused weighting parameters

Browse files

Removes 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.

Files changed (1) hide show
  1. train_pnet_class_cluster.py +1 -1
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, score_weight=1.0, class_weight=0.5)
 
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)