octree-nn / ocnn-pytorch

Octree-based Sparse Convolutional Neural Networks
MIT License
150 stars 16 forks source link

Pre-trained model weights? #16

Closed harryseely closed 1 year ago

harryseely commented 1 year ago

I am interested in using transfer learning for OCNN-LeNet. Would you be able to provide a pre-trained model (e.g., .t7, .pth) file? I can't seem to find one online...

wang-ps commented 1 year ago

The pretrained models can be found in the documentation: https://ocnn-pytorch.readthedocs.io/en/latest/index.html

image

harryseely commented 1 year ago

Amazing thank you!