"I found a problem. On line 47 of Vim/seg/backbone/vim.py, the parameters are not mapped correctly. num_classes is mapped to in_chans."