search

moabitcoin / ig65m-pytorch

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

Adds hubconf.py for pytorch hub integration, resolves #4 #23

Closed daniel-j-h closed 5 years ago

daniel-j-h commented 5 years ago

Adds PyTorch Hub integration

https://pytorch.org/docs/stable/hub.html

Once this lands users can simply say:

import torch

model = torch.hub.load("moabitcoin/ig65m-pytorch", "r2plus1d_34_32_ig65m",
                       num_classes=359, pretrained=True)

Ran into https://github.com/pytorch/pytorch/issues/27844 on the way; but works on our end now.