Train

This is where we train a point cloud classification model
 using PyTorch Lightning and MinkowskiEngine.

We first set up DataLoader instances for the training and validation datasets,
 and configures a PyTorch Lightning Trainer instance to manage the training process,
 then saves the trained model.

The training will stop if the validation loss does not improve after a
 specific number of epochs, the best model will be saved.


Functions:
  train_main(model, model_save_path, num_workers): 
    The main function to train the model, taking the model instance,
    the path to save the trained model, and the number of workers to use
    for data loading.




Last updated