Loading base network...
Traceback (most recent call last):
File "train_test.py", line 133, in
net.base.load_state_dict(base_weights)
File "/home/xxxxx/anaconda2/envs/pyssd/lib/python3.6/site-packages/torch/nn/modules/module.py", line 490, in load_state_dict
.format(name))
KeyError: 'unexpected key "0.weight" in state_dict'
Error while training RBF_mobile version
Loading base network... Traceback (most recent call last): File "train_test.py", line 133, in
net.base.load_state_dict(base_weights)
File "/home/xxxxx/anaconda2/envs/pyssd/lib/python3.6/site-packages/torch/nn/modules/module.py", line 490, in load_state_dict
.format(name))
KeyError: 'unexpected key "0.weight" in state_dict'