From 40bb0597bb03aa3ea7ea758ab146e2a9264a0060 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 18 Aug 2020 08:50:24 -0600 Subject: [PATCH] misc --- codes/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/train.py b/codes/train.py index 97963dcb..fd21808a 100644 --- a/codes/train.py +++ b/codes/train.py @@ -32,7 +32,7 @@ def init_dist(backend='nccl', **kwargs): def main(): #### options parser = argparse.ArgumentParser() - parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_imgset_spsr_switched2_wavelet.yml') + parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_mi1_spsr_switched2.yml') parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher') parser.add_argument('--local_rank', type=int, default=0)