config.py

import tensorkit as tk

config = tk.Config()
_config = {
    'TRAIN_DS': '/data2/jaep0805/datasets/samsungdataset_name_changed/CVPR2020_NTIRE_Workshop/train',
    'TRAIN_SIZE': 85,
    'VAL_DS': './dataset/kalantari_dataset/val',
    'VAL_SIZE': 55,
    'VAL': False,
    'TEST_DS': '/data2/jaep0805/datasets/samsungdataset_name_changed/CVPR2020_NTIRE_Workshop/test',
    'TEST_SIZE': 15,

    'train_hw': (512, 512),  # The size of images for training
    'val_hw': (256, 256),
    'test_hw': (960, 1440),

    'BATCH_SIZE': 4,
    'EPOCH': 256000,

    'LR': 1e-4,
    'LR_DECAY': 0.92,  # per epoch from 90 epoch to 120 epoch
    'MOMENTUM': 0.9,

    'LOG_DIR': './logs',
    'TEST_DIR': './results',

    'SAVE_STEP': 1000,
    'SUMMARY_STEP': 200,
    
    # 'SAVE_STEP': 100,
    # 'SUMMARY_STEP': 100,

    'CUDA_VISIBLE_DEVICES': [0],
    'ALLOW_GROWTH': False,

    'BACKUP_DIR': ['model', 'tensorkit'],
    'VGG_CKPT': './model_zoo/vgg_16.ckpt',
}

config.apply(_config)

if __name__ == '__main__':
    config.show()