Co-DETR icon indicating copy to clipboard operation
Co-DETR copied to clipboard

作者您好,如何启动AMP进行训练呢

Open dhddxdhd opened this issue 1 year ago • 4 comments

我的电脑现存太小了

dhddxdhd avatar Oct 30 '23 02:10 dhddxdhd

或者说在您的代码上,有其他降低显存并且对精度影响不大的方法吗,恳请大佬指点

dhddxdhd avatar Oct 30 '23 02:10 dhddxdhd

同问,如何在不影响精度的情况下降低现存呢,训练速度倒是无所谓

ocrhei avatar Oct 30 '23 03:10 ocrhei

目前官方mmdetection 3.2已经支持了Co-DETR算法,关于大模型训练减少显存可以参考https://github.com/open-mmlab/mmdetection/tree/main/projects/example_largemodel

TempleX98 avatar Oct 30 '23 07:10 TempleX98

请问直接在您的代码上可以减少显存么,,哎,mmdetection3系列看不懂

6098669 avatar Oct 30 '23 11:10 6098669