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

floodsung / LearningToCompare_FSL

PyTorch code for CVPR 2018 paper: Learning to Compare: Relation Network for Few-Shot Learning (Few-Shot Learning part)

MIT License

1.04k stars 268 forks source link

Open huajie1112 opened 4 years ago

huajie1112 commented 4 years ago

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