关于人工智能:YOLOX-train训练提示CUDA-out-of-memory

将YOLOX/yolox/exp/yolox_base.py中的
self.data_num_workers = 4
更改为:
self.data_num_workers = 0
本人测试的时候,--fp16-o并没有什么影响。

评论

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

这个站点使用 Akismet 来减少垃圾评论。了解你的评论数据如何被处理