我正在尝试训练一个YOLOv5神经网络来识别车辆。然而,当它在Google Colab上训练时,它总是停留在这里:
train: Scanning 'MyDataset/train/labels.cache' for images and labels... 26559 found, 0 missing, 0 empty, 0 corrupted: 100% 26559/26559 [00:00<?, ?it/s]
train: Caching images (8.5GB): 62% 16425/26559 [00:46<00:30, 330.41it/s]C
CPU times: user 850 ms, sys: 162 ms, total: 1.01 s
Wall time: 1min 26s我遵循了roboflow的教程。当我切换到roboflow提供的较小的数据库时,训练能够继续进行。我是Colab Pro+的用户,所以这不应该是内存不足的问题。
发布于 2021-11-18 03:38:03
我切换到一个较小的数据集,现在它加载没有任何问题。
train: Caching images (4.6GB): 100% 8853/8853 [00:18<00:00, 483.20it/s]然后它顺利地开始了训练。我认为这确实是数据太多的问题。然而,Colab没有给我任何内存不足的迹象。
https://stackoverflow.com/questions/70014098
复制相似问题