Fewshot-Learning-with-BERT

Fewshot-Learning-with-BERT copied to clipboard

 the first image say:generated batch data  the first image say:generated lable i think they dont matched!

majinmin

question about 1/5 shot

1

Hi, I am interested in your work and have tried to run train.py. Is it the result with configuration of 1-shot? When I change "Nc" to 5, I get accuracy...

venusafroid

train.py程序中计算准确率使用的label是随机生成的。 个人认为应该调用模型生成的label与真实label进行比较，而非随机生成的label与真实label比较，该问题导致模型训练准确率一直在0.5左右 label = torch.arange(5).repeat(10).type(torch.LongTensor).cuda() correct += float(torch.sum(logits == label).item())

zgf0907