multiyolov5
multiyolov5 copied to clipboard
训练时遇到问题,提示如下
❔Question
作者你好,请问我在训练的时候train_custom.py的时候出现错误File "train_custom.py", line 641, in
Additional context
你把batch_size改小一点就行了,因为这个训练集数据少,batch_size默认为16
谢谢了,确实是这个问题,问题解决 @hfutwqs
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.