This repository was archived by the owner on Jun 13, 2024. It is now read-only.
This repository was archived by the owner on Jun 13, 2024. It is now read-only.
Hello, I trained the multi model, but when i load the weights file i get an error，please how to solve this problem? #182
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