RuntimeError: shape '[-1, 128, 19, 19]' is invalid for input of size 46656000 on miniimagenet code

[Alan11211]

Hello, I am trying to execute the miniimagenet code on different dataset (colored dataset). I have customized the batch number per class to 5, and resized the input images to 84*84. besides these I have not made any changes to code.
can you please help Why I am getting this error and how it can be fixed ?

Traceback (most recent call last):
File "miniimagenet_train_one_shot.py", line 275, in
main()
File "miniimagenet_train_one_shot.py", line 192, in main
relation_pairs = torch.cat((sample_features_ext,batch_features_ext),2).view(-1,FEATURE_DIM*2,19,19)
RuntimeError: shape '[-1, 128, 19, 19]' is invalid for input of size 46656000