snail-pytorch

snail-pytorch copied to clipboard

Some discrepancies from the paper

2

For the Mini-ImageNet ResNet-based encoder, I believe the authors use mean pooling instead of max pooling, as done by this implementation. I am currently unable to reproduce the mini-imagenet results...

ericjang

Hi have you finished the RL part? thanks in advance!!

Wongcheukwai

random uniform choosing of k in k-shot in training

4

In the original paper, "the number of shots K was chosen uniformly at random from 1 to 5." So the forward function of SNAIL should take an arbitrary K instead...

victorychain