diff --git a/codes/train.py b/codes/train.py index 2b543e99..91c8f75f 100644 --- a/codes/train.py +++ b/codes/train.py @@ -293,7 +293,7 @@ class Trainer: if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_xx_faces_glean.yml') + parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_faces_styled_sr.yml') parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher') parser.add_argument('--local_rank', type=int, default=0) args = parser.parse_args() @@ -306,7 +306,7 @@ if __name__ == '__main__': print('export CUDA_VISIBLE_DEVICES=' + gpu_list) trainer = Trainer() -#### distributed training settings + #### distributed training settings if args.launcher == 'none': # disabled distributed training opt['dist'] = False trainer.rank = -1 @@ -315,7 +315,7 @@ if __name__ == '__main__': print('Disabled distributed training.') else: opt['dist'] = True - init_dist('nccl', opt) + init_dist('nccl') trainer.world_size = torch.distributed.get_world_size() trainer.rank = torch.distributed.get_rank() diff --git a/codes/trainer/eval/style.py b/codes/trainer/eval/fid.py similarity index 100% rename from codes/trainer/eval/style.py rename to codes/trainer/eval/fid.py