from fast_pointnet_v2 import train_pointnet

if __name__ == "__main__":

    # Load the dataset
    dataset_path = "path/to/your/dataset"
    model_save_path = "path/to/your/model.pth"

    # Train the model
    train_pointnet(dataset_path, model_save_path, epochs=100, batch_size=512, lr=0.001, score_weight=0.25, class_weight=1.0)