Also got an error at line assert dshape[1] % cfactor == 0 of _pack_weight with a MobileNetV2-based NN made with PyTorch. When playing with start_name and stop_name and their indexes, it appears that some layers have Tensor sizes of [a, 1, b, c], causing the assert to fail. Those layers seem to be the bottleneck layers’ middle convolution layers…
Would implementing a assert dshape[1] % cfactor != 0 case really be the only way to fix this issue?
Has anyone else tried MobileNetV2 with a VTA graph_pack by any chance?
Thanks.