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