search

yaohungt / Gated-Spatio-Temporal-Energy-Graph

[CVPR'19] [PyTorch] Gated Spatio Temporal Energy Graph
151 stars 25 forks source link

Problem with the pretrained weights #10

Closed sinaazar closed 4 years ago

sinaazar commented 4 years ago

Hi,

I tried to use the i3d implementation here for another task. However, it seems like the provided weights are not just from training on Kinetics. I guess it was also trained on charades. this is the error when I try to load weights for a model to classify Kinetics (as an example). size mismatch for logits.conv3d.weight: copying a param with shape torch.Size([157, 1024, 1, 1, 1]) from checkpoint, the shape in current model is torch.Size([400, 1024, 1, 1, 1]) Is it possible to include Kinetics-only weights?

Thanks

yaohungt commented 4 years ago

There are a lot of i3d pre-trained models for Kinetics online.

sinaazar commented 4 years ago

I see. I preferred this implementation. Anyways, the Readme is misleading. You might want to edit it.

yaohungt commented 4 years ago

OK, Thx