Hello, I trained the multi model, but when i load the weights file i get an error，please how to solve this problem?

[SiaLight]

the function load_weights and load_weights_until_last has the error

RuntimeError: invalid argument 2: size '[512 x 256 x 3 x 3]' is invalid for input with 63516 elements at /opt/conda/conda-bld/pytorch_1535491974311/work/aten/src/TH/THStorage.cpp:84