search

Haochen-Wang409 / U2PL

[CVPR'22] Semi-Supervised Semantic Segmentation Using Unreliable Pseudo-Labels
Apache License 2.0
426 stars 59 forks source link

显存溢出问题 #81

Closed yang654123 closed 2 years ago

yang654123 commented 2 years ago
image

你好，作者，我使用8张V100。16G 。保留您的参数不变，结果还是出现了爆显存的问题，请问，这种情况如何应对呢

yang654123 commented 2 years ago

我调低了batch size为2 可以运行了，但是应该跑不出您的效果了

Haochen-Wang409 commented 2 years ago

同步调整 learning rate 即可。请仔细阅读 readme 中 note 的部分

yang654123 commented 2 years ago

同步调整 learning rate 即可。请仔细阅读 readme 中 note 的部分

请问同步调整是调整为为多少合适呢，您可否推荐一个，十分感谢，我的batchsize=2

Haochen-Wang409 commented 2 years ago

几张卡？跑的是VOC还是CS？是SupOnly还是Semi？

yang654123 commented 2 years ago

几张卡？跑的是VOC还是CS？是SupOnly还是Semi？

8张V100 VOC semi

Haochen-Wang409 commented 2 years ago

不需要调整学习率，直接训练即可。

我们的 config 是 4x4 的 batch_size，你采用的是 2x8，总的都是 16

yang654123 commented 2 years ago

不需要调整学习率，直接训练即可。

我们的 config 是 4x4 的 batch_size，你采用的是 2x8，总的都是 16

好的 感谢