TransFG
TransFG copied to clipboard
About train.py
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 ?
I also use that,but it didn't report an UserWarning like this.