diff --git a/configs/restorers/ttsr/ttsr-rec_x4_c64b16_g1_200k_CUFED.py b/configs/restorers/ttsr/ttsr-rec_x4_c64b16_g1_200k_CUFED.py index ff36470712..ea5d52c79d 100644 --- a/configs/restorers/ttsr/ttsr-rec_x4_c64b16_g1_200k_CUFED.py +++ b/configs/restorers/ttsr/ttsr-rec_x4_c64b16_g1_200k_CUFED.py @@ -223,7 +223,7 @@ scale=scale)) # optimizer -optimizers = dict(type='Adam', lr=1.e-4) +optimizers = dict(generator=dict(type='Adam', lr=1.e-4)) # learning policy total_iters = 200000