请问一下您运行时的gpu条件是？

[yyzih]

这边尝试在自己的机子上跑您的代码，2080ti 11GB，但只有5way-1shot的ConvNet能跑起来，5shot或者是ResNet12都会出现类似
RuntimeError: CUDA out of memory. Tried to allocate 36.00 MiB (GPU 0; 10.76 GiB total capacity; 9.33 GiB already allocated; 56.69 MiB free; 219.71 MiB cached)
的问题。想知道您这边运行是的gpu是？看了下您的论文也没有提到gpu这块。

[heng-yin]

可以试一下调小batch size。

[zilunzhang]

这边尝试在自己的机子上跑您的代码，2080ti 11GB，但只有5way-1shot的ConvNet能跑起来，5shot或者是ResNet12都会出现类似 RuntimeError: CUDA out of memory. Tried to allocate 36.00 MiB (GPU 0; 10.76 GiB total capacity; 9.33 GiB already allocated; 56.69 MiB free; 219.71 MiB cached) 的问题。想知道您这边运行是的gpu是？看了下您的论文也没有提到gpu这块。

你好，

1shot的convnet和resnet12是用的2080ti单卡训练，resnet18和wrn以及5shot基本是v100。如果显存不够可尝试调小batchsize或者多卡来跑（num_gpu需要调整）