search

Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
1.01k stars 111 forks source link

作者您好，如何启动AMP进行训练呢 #80

Open dhddxdhd opened 1 year ago

dhddxdhd commented 1 year ago

我的电脑现存太小了

dhddxdhd commented 1 year ago

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

ocrhei commented 1 year ago

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

TempleX98 commented 1 year ago

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

6098669 commented 1 year ago

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