search

yjxiong / tsn-pytorch

Temporal Segment Networks (TSN) in PyTorch
BSD 2-Clause "Simplified" License
1.07k stars 308 forks source link

Parameter_load #42

Closed bupt-zsp closed 6 years ago

bupt-zsp commented 6 years ago

If I don't want to load the BN-Inception pre-train weight, and set the random weight on BN-Inception. what should I do, just annotation the load_state_dict function? thx~

yjxiong commented 6 years ago

comment out this line https://github.com/yjxiong/tensorflow-model-zoo.torch/blob/master/bninception/pytorch_load.py#L35