SenHe / Flow-Style-VTON

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

learning rate

tengolagan opened this issue · 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) ?