kenshohara / video-classification-3d-cnn-pytorch

Video classification tools using 3D ResNet
MIT License
1.11k stars 260 forks source link

Load model #19

Open phiwei opened 6 years ago

phiwei commented 6 years ago

Hi, I am relatively new to PyTorch but have a data set that would be a nice use case for your model. What is the easiest way for me to load a pretrained model as feature extractor? Like say I would want the resnet10 pretrained, how could I get that into my code?

Sorry for bothering you with this, I am sure this is self-evident to more experienced programmers. If it performs well on my data set I will let you know and cite of course :)