floodsung/LearningToCompare_FSL

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 2. Got 1 and 50 in dimension 0 at /pytorch/torch/lib/THC/generic/THCTensorMath.cu:111

huajie1112 opened this issue · 0 comments

When I run the python miniimagenet_train_few_shot.py, it doesnot work. I don't know how to fix it.