Yolo model training on colab got errors
633
0
07-09-2021 11:17 PM
by
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()
