Flow-Style-VTON icon indicating copy to clipboard operation
Flow-Style-VTON copied to clipboard

learning rate

Open tengolagan opened this issue 1 year ago • 0 comments

How come ResUnetGenerator doesn't have a method update_learning_rate?

Last 3 rows of code in train_PBAFN_e2e_fs.py :

Capture d’écran 2023-04-13 à 22 38 08

Why are we updating model (the warping model) twice ? Shouldn't model.module.update_learning_rate(optimizer_gen) be something like model_gen.module.update_learning_rate(optimizer_gen) ?

tengolagan avatar Apr 13 '23 20:04 tengolagan