machine-learning - Colab 无法加载缓存

Question

我正在尝试训练一个 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+ 用户，所以不应该是内存不足的问题。

Answer 1

我切换到一个较小的数据集，现在它加载没有任何问题。

train: Caching images (4.6GB): 100% 8853/8853 [00:18<00:00, 483.20it/s]

然后就顺利的开始训练了。我认为这确实是数据过多的问题。但是 Colab 并没有给我任何内存不足的迹象。