Tiny YOLOv4 cfg modification to use sgdr #7636
-
Hi I have been modifying yolov4-tiny-custom.cfg to use the policy sgdr when training TinyTolov4 using an aerial car dataset. The thing is I am not able to achieve more than 1500 iterations when actually the max_batches is set to 3990. If someone could help me I would really appreciate it. The error outputted from the terminal of Google colab is the following: CUDA Error: an illegal memory access was encountered And the modified parts are in the [net] layer: Trainingbatch=64 learning_rate=0.0028 |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
Fixed by addinging <steps=3200,3600> below the policy. |
Beta Was this translation helpful? Give feedback.
Fixed by addinging <steps=3200,3600> below the policy.