search

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

Update omniglot_train_few_shot.py #32

Open Bryce1010 opened 4 years ago

Bryce1010 commented 4 years ago

update the pytorch about "data[0]" to item()