ESTRNN/main.py at master · zzh-tech/ESTRNN · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from para import Parameter
from train import Trainer

if __name__ == '__main__':
    para = Parameter().args
    # para.model = 'ESTRNN'

    # # choose one dataset
    # para.dataset = 'BSD'
    # para.dataset = 'gopro_ds_lmdb'
    # para.dataset = 'reds_lmdb'
    # para.video = True

    # # resume training from existing checkpoint
    # para.resume = True
    # para.resume_file = './experiment/2020_12_29_01_56_43_ESTRNN_gopro_ds_lmdb/model_best.pth.tar'

    # # test using existing checkpoint
    # para.test_only = True
    # para.test_save_dir = './results/'
    # para.test_checkpoint = './experiment/2020_12_29_01_56_43_ESTRNN_gopro_ds_lmdb/model_best.pth.tar'

    trainer = Trainer(para)
    trainer.run()