search

zhoubolei / TRN-pytorch

Temporal Relation Networks
http://relation.csail.mit.edu/
Other
787 stars 190 forks source link

RuntimeError： loading_state_dict for BNInception #31

Open cvnovice95 opened 5 years ago

cvnovice95 commented 5 years ago

size mismatch for conv1_7x7_s2_bn.weight: copying a param of torch.Size([64]) from checkpoint, where the shape is torch.Size([1, 64]) in current model. size mismatch for conv1_7x7_s2_bn.bias: copying a param of torch.Size([64]) from checkpoint, where the shape is torch.Size([1, 64]) in current model.

gamzeakyol commented 5 years ago

I encountered with same issue, but could not solve it. Pytorch version is 0.4.1. Opening a new conda environment with Pytorch 0.3 was not also worked.

cvnovice95 commented 5 years ago

use pytorch 0.3.1