Convert i3d_nonlocal_8x8_IN_pretrain_400k

Tushar-N / pytorch-resnet3d

I3D Nonlocal ResNets in Pytorch

245 stars 39 forks source link

Convert i3d_nonlocal_8x8_IN_pretrain_400k #10

Closed ShiroKL closed 4 years ago

ShiroKL commented 4 years ago

Hi,

I was wondering if it is possible with your code to convert i3d_nonlocal_8x8_IN_pretrain_400k model and the c2d model from the facebook repository ?

EDIT : The conversion "works" there is no error during the conversion but I can not check if the performance are the same because I do not have the kinetics dataset.

Tushar-N commented 4 years ago

I've hard-coded params for the two models, so it likely won't work. You'd need to adjust the parameters or write a more general version of the conversion/model scripts.

ruiyan1995 commented 4 years ago

@ShiroKL Did you have convert c2d model from caffe to pytorch?