From a93167d92f7e6e7811a24d510793966c964f4845 Mon Sep 17 00:00:00 2001 From: Liu Changyu Date: Wed, 17 Nov 2021 10:21:44 +0800 Subject: [PATCH] Update config.py --- config.py | 27 ++++++++++++++++----------- 1 file changed, 16 insertions(+), 11 deletions(-) diff --git a/config.py b/config.py index f6eb3d1..0920711 100644 --- a/config.py +++ b/config.py @@ -31,8 +31,13 @@ if mode == "train": # Dataset # Image format - train_image_dir = f"data/T91_General100/train" - valid_image_dir = f"data/T91_General100/valid" + train_image_dir = f"data/TG191/train" + valid_image_dir = f"data/TG191/valid" + # LMDB format + train_lr_lmdb_path = f"data/train_lmdb/TG191_LRbicx{upscale_factor}_lmdb" + train_hr_lmdb_path = f"data/train_lmdb/TG191_HR_lmdb" + valid_lr_lmdb_path = f"data/valid_lmdb/TG191_LRbicx{upscale_factor}_lmdb" + valid_hr_lmdb_path = f"data/valid_lmdb/TG191_HR_lmdb" image_size = 72 batch_size = 16 @@ -43,20 +48,20 @@ start_epoch = 0 resume_weight = "" - # Total number of epochs + # SGD optimizer use: . Adam optimizer use: . epochs = 600 # Model optimizer parameter (less training and low PSNR) - # model_optimizer_name = "sgd" - # model_lr = 1e-3 - # model_momentum = 0.9 - # model_weight_decay = 1e-7 - # model_nesterov = False + model_optimizer_name = "sgd" + model_lr = 1e-3 + model_momentum = 0.9 + model_weight_decay = 1e-7 + model_nesterov = False # Modify optimizer parameter (faster training and better PSNR) - model_optimizer_name = "adam" - model_lr = 1e-3 - model_betas = (0.9, 0.999) + # model_optimizer_name = "adam" + # model_lr = 1e-3 + # model_betas = (0.9, 0.999) # ============================================================================== # Verify configuration