search

gitabcworld / MatchingNetworks

This repo provides pytorch code which replicates the results of the Matching Networks for One Shot Learning paper on the Omniglot and MiniImageNet dataset
Other
328 stars 81 forks source link

Double softmax #5

Open bkj opened 5 years ago

bkj commented 5 years ago

On this line you're applying a softmax to the similarities.

Then later you apply cross_entropy, which is a log softmax + NLL loss.

I think you probably want to remove the first softmax.

~ Ben