Spaces:
Paused
Paused
Baseline Model using PyTorch Lightning and ONNX
#15
by
bvk1ng
- opened
For those interested in using PyTorch Lightning for this competition, I have created and shared a notebook on Kaggle:
https://www.kaggle.com/dextermojo/eda-baseline-hf-competetion-pytorchlightning
showcasing the use of PyTorch Lightning for training a baseline model on the data and then using ONNX for model inferencing.
If you find the notebook informative, please give it an upvote.
Cheers,
Adi
Very nice!
Interesting use of the Random Rain augmentation. Is that a secret sauce trick?
Haha, sorry for the late reply. I don't know the exact answer to your question as I have tried multiple augmentations, but it could be one of the factors.