TACJu/TransFG

About train.py

Christine620 opened this issue · 1 comments

I found you write scheduler.step() before optimizer.step() in line 267-268 in train.py, when I run it,i got a UserWarning UserWarning: Seems like optimizer.step() has been overridden after learning rate scheduler initialization. Please, make sure to call optimizer.step() before lr_scheduler.step().,so is your code right on that place ?

cwq63 commented

I also use that,but it didn't report an UserWarning like this.