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

61次阅读

共计 107 个字符,预计需要花费 1 分钟才能阅读完成。

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

正文完
 0