Closed yt605155624 closed 1 year ago
The scheduler was copied directly from StarGANv2, including the learning rate of the mapping network being 1e-6. You can refer to https://github.com/clovaai/stargan-v2 for more details on the settings. I didn't change any hyperparameters here, so they were hardcoded.
mapping_network 一直是 2e-6,其他一直是 2e-4 因为 div_factor 和 final_div_factor 都是 1,那我理解是否无需设定 scheduler 了,直接所有的 learing_rate 都在 AdamW 里面写死为 2e-4 或者 2e-6 即可?