moabitcoin / ig65m-pytorch

PyTorch 3D video classification models pre-trained on 65 million Instagram videos
MIT License
265 stars 30 forks source link

Provide convenient fine-tuning tool for users #7

Open daniel-j-h opened 5 years ago

daniel-j-h commented 5 years ago

We should provide a tool where the user can load up a pre-trained model, point it at a dataset (e.g. directory of labeled videos), and have it fine-tune the model automatically.

bjuncek commented 5 years ago

you can easily repurpose the reference script from torchvision for that :)

daniel-j-h commented 5 years ago

Good point! For reference:

https://github.com/pytorch/vision/tree/v0.4.0/references/video_classification