请问实验环境使用了几张什么型号的显卡？

[qq31415926]

请问实验环境使用了几张什么型号的显卡？

[lyyang]

你好，1-shot的实验设置下用了1张V100，5-shot的实验设置下用了2~3张V100

[Hj]

你好，请问，您的显卡使用中，哪一部分占用的内存最大呢？

[lyyang]

你好，占用内存比较大的地方主要是“每个类别下取多少个样本”，也就是取多少shot，我这里使用的V100的每个显存为32G。这里对于5way1shot的小样本学习的定义不是只拿5个样本来进行训练，而是从原始的大数据集中采样很多这种5way1shot的sample进行训练，因为测试集中的类别和训练集中的类别是不重合的。

[lyyang]

你好，请问，您的显卡使用中，哪一部分占用的内存最大呢？

具体数据的采样过程可以看data_loader.py的FewRelDataset函数部分，参数上要保证Q_na=0，否则会采样进来多的样本。