diff --git a/main_train_msrresnet_gan.py b/main_train_msrresnet_gan.py index 65e5dfcd..22aaf017 100644 --- a/main_train_msrresnet_gan.py +++ b/main_train_msrresnet_gan.py @@ -125,8 +125,8 @@ def main(json_path='options/train_msrresnet_gan.json'): model = define_Model(opt) - logger.info(model.info_network()) model.init_train() + logger.info(model.info_network()) logger.info(model.info_params()) '''