cuda out of memory

[wwencc]

当shot=5时，训练coco数据集时显示显存不足。

[qjmltz]

您好，可否发我一份HDMNet的数据列表呢？项目中的文件链接已经失效了， 还有就是我试着用了一下BAM的lists，发现显示 HDMNet\train.py", line 122, in main assert args.zoom_factor in [1, 2, 4, 8] HDMNet\util\config.py", line 30, in getattr raise AttributeError(name) AttributeError: zoom_factor 不知道您有没有遇见过，是我的config设置的不对吗？

[wwencc]

在config文件中增加一行代码，设置参数zoom_factor。
manual_seed: 321 seed_deterministic: False zoom_factor: 8 # zoom factor for final prediction during training, be in [1, 2, 4, 8]

[qjmltz]

在config文件中增加一行代码，设置参数zoom_factor。 manual_seed: 321 seed_deterministic: False zoom_factor: 8 # zoom factor for final prediction during training, be in [1, 2, 4, 8]

非常感谢您的回复，前两天我做了复现，因为作者给的lists和预训练模型链接都失效了，我直接使用BAM项目中的lists和预训练模型，仅复现了pascal数据集1 shot vgg预训练模型下的flod0-3，但得到的mIoU差值都比较大，之前在BAM中只有千分位的误差。请问您遇到过这样的问题吗？我担心是我在yaml中参数没有设置好

[wwencc]

我没有使用vgg，而是使用resnet50网络复现了pascal数据集，得到的4折的最终平均mIoU与论文差不多，每一折的mIou和论文的数据有一些差距。