To update this post:


The same code works with epochs = 9, batch_size = 16 and not using the Yolo weights (from scratch) and with less data (around 800 images instead of 10k)


But this code doesn’t work with epoch = 9, batch_size = 16 and not using the Yolo weight, if I keep all the dataset (it return the same CUDA error)


It’s very weird …