Issue about input dimension of MultiscaleresNet18

[lam1051999]

Dear sir @JinghuiZhou, thanks for a great repository and many network architecture implementations. But i have an issue that i think the code need to do a little bit more configuration. When i was training Resnet18, Resnet30, VGG,... everything is ok. But when i switch to MultiscaleresNet18, i got a bug below

RuntimeError: Expected 4-dimensional input for 4-dimensional weight [64, 3, 7, 7], but got 5-dimensional input of size [2, 1, 3, 224, 224] instead

The error happens in the line summary(model, (3, 224, 224)) and the input of the model in main.py.
I think the code need a small change when training with MultiscaleresNet. But i couldn't find where to fix it. Can you show me how? Thanks a lot!!!