search

zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
GNU Lesser General Public License v3.0
5.2k stars 1.27k forks source link

train problem in ubuntu #733

Closed diplomatist closed 1 year ago

diplomatist commented 1 year ago

image 在ubuntu系统训练一直卡在这个地方，但我在windows下能训练，这是为什么？无论我将num_worker设为0，num_gpu设为1都不行。

diplomatist commented 1 year ago

image 我debug了train.py，发现在训练开始迭代的时候，一直显示Collecting data...卡在这了。这该怎么解决

diplomatist commented 1 year ago

image

diplomatist commented 1 year ago

我解决了这个问题，内存容量太小导致