cfg_preprocessed.py

import os
import time

#data config
input_dir = '/media/gtmeng/DataDisk2/Learning-to-See-in-the-Dark-pre-processed/VBM4D_rawRGB'
gt_dir = '/media/gtmeng/DataDisk2/Learning-to-See-in-the-Dark-pre-processed/long'
train_crop_size=256


#train config
input_channels=3
seed=1
train_batch_size=1
data_loader_num_workers=4
base_lr=1e-4
momentum=0.9
total_epochs=1000
lr_decay_epochs=[500]
lr_decay_rate=0.1
model_save_interval=50
model_save_path='./snapshots_preprocessed'
vgg_output_layer_list=[2,7,16,25]
loss_c_ratio=0.05

if not os.path.exists(model_save_path):
    os.mkdir(model_save_path)

log_interval=4
time_str=time.asctime( time.localtime(time.time()))
log_dir=os.path.join('logs/runs_preprocessed','-'.join(time_str.split()[1:-1]))

demosaic=False

start_epoch=1
for epoch in lr_decay_epochs:
    if epoch<=start_epoch:
        base_lr*=lr_decay_rate

if start_epoch>1:
    snapshot_path=os.path.join(model_save_path,'model_%05d.pth'%(start_epoch-1))
else:
    snapshot_path=None