运行yolov5时会提示In PyTorch 1.1.0 and later, you should call them in the opposite order: `optimizer.step()` before `optimizer.step()`.CSDN有大佬讲了具体原因和操作,但没有针对ylov5 v6.2版本的修改方法。具体方法就是修改train.py

# Scheduler
lr = [x['lr'] for x in optimizer.param_groups]  # for loggers
scheduler.step()

移动scheduler.step()到下面的位置,注意与end training前面的#对齐

    # end epoch ----------------------------------------------------    
# end training ------------------------------------------------------------------------

下面就是修改后的


        # end epoch -----------------------------------------------------------------------
    scheduler.step()
    # end training ------------------------------------------------------------------------

更多推荐