VMZ. Pytorch models

Hi followed the install instructions to get vmz locally.

I wanted to try out some of the pretrained models like

$ from vmz.models import r2plus1d_34 $ r2plus1d_34(pretraining=‘sports1m_32frms’)

posted below is the tail output of the model: (avgpool): AdaptiveAvgPool3d(output_size=(1, 1, 1)) (fc): Linear(in_features=512, out_features=400, bias=True) this seems like its the finetuned weights on kinetics 400 using pretrained weights, as the fc output should have been 487 from Sports1m instead of 400. I figure this is a bug in matching args to urls in . utils.py

similarly, you have key mismatch running $ r2plus1d_34(pretraining=‘ig65m_32frms’)

looking up the model_urls at line 7 of utils.py it works when i run with

$ r2plus1d_34(pretraining=‘65m_32frms’) ) ) (avgpool): AdaptiveAvgPool3d(output_size=(1, 1, 1)) (fc): Linear(in_features=512, out_features=400, bias=True) ) I might be off since I am new to video CNN’s, just wanted to point this out. I am interested in using pretrained weights from sport1m and IG65m on my own video datasets using newer CNN’s like r2plus1d_34

[dutran]

we have R(2+1)D-34 preatrained on Sports-1M but on 32x112x112 not 224x224 which may have worse performance, you can find them here. https://github.com/facebookresearch/VMZ/blob/master/c2/tutorials/models.md

[bjuncek]

Ah, yes - I believe that was one of the internal ones.

@dutran do you perhaps have the R(2+1)D-34 pretrained on sports1m only lying around somewhere? The one you’ve later finetuned for the 2017/8 CVPR paper in models.md.