Yolo model training on colab got errors

211
0
07-09-2021 11:17 PM
KyawNaingWin
New Contributor II

After successfully trained SSD model on Colab, I try Yolov3 training. This time I got error at early stage.

I  do not get such error on my windows laptop but since I do not have GPU, I want to train it on Colab.

the message said "/usr/local/lib/python3.7/site-packages/arcgis/learn/models/_yolov3_utils.py ....RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead."

model.lr_find()
delete_later.JPG
0 Kudos
0 Replies