2020-1-capstone-design2 / 2016104167

Go to a project
Toggle navigation Toggle navigation pinning
조현아
9c71da6c 1 parent 75bcfcee

train max_step_6500

Expand all
Inline Side-by-side
Showing 2 changed files with 6 additions and 6 deletions
This diff is collapsed. Click to expand it.
...@@ -104,20 +104,20 @@ def dict_to_namedtuple(d): ...@@ -104,20 +104,20 @@ def dict_to_namedtuple(d):
104 104
105 def parse_args(kwargs): 105 def parse_args(kwargs):
106 # combine with default args 106 # combine with default args
107 - kwargs['dataset'] = kwargs['dataset'] if 'dataset' in kwargs else 'cifar10' 107 + kwargs['dataset'] = kwargs['dataset'] if 'dataset' in kwargs else 'BraTS'
108 - kwargs['network'] = kwargs['network'] if 'network' in kwargs else 'resnet_cifar10' 108 + kwargs['network'] = kwargs['network'] if 'network' in kwargs else 'resnet50'
109 kwargs['optimizer'] = kwargs['optimizer'] if 'optimizer' in kwargs else 'adam' 109 kwargs['optimizer'] = kwargs['optimizer'] if 'optimizer' in kwargs else 'adam'
110 - kwargs['learning_rate'] = kwargs['learning_rate'] if 'learning_rate' in kwargs else 0.1 110 + kwargs['learning_rate'] = kwargs['learning_rate'] if 'learning_rate' in kwargs else 0.0001
111 kwargs['seed'] = kwargs['seed'] if 'seed' in kwargs else None 111 kwargs['seed'] = kwargs['seed'] if 'seed' in kwargs else None
112 kwargs['use_cuda'] = kwargs['use_cuda'] if 'use_cuda' in kwargs else True 112 kwargs['use_cuda'] = kwargs['use_cuda'] if 'use_cuda' in kwargs else True
113 kwargs['use_cuda'] = kwargs['use_cuda'] and torch.cuda.is_available() 113 kwargs['use_cuda'] = kwargs['use_cuda'] and torch.cuda.is_available()
114 kwargs['num_workers'] = kwargs['num_workers'] if 'num_workers' in kwargs else 4 114 kwargs['num_workers'] = kwargs['num_workers'] if 'num_workers' in kwargs else 4
115 - kwargs['print_step'] = kwargs['print_step'] if 'print_step' in kwargs else 2000 115 + kwargs['print_step'] = kwargs['print_step'] if 'print_step' in kwargs else 500
116 - kwargs['val_step'] = kwargs['val_step'] if 'val_step' in kwargs else 2000 116 + kwargs['val_step'] = kwargs['val_step'] if 'val_step' in kwargs else 500
117 kwargs['scheduler'] = kwargs['scheduler'] if 'scheduler' in kwargs else 'exp' 117 kwargs['scheduler'] = kwargs['scheduler'] if 'scheduler' in kwargs else 'exp'
118 kwargs['batch_size'] = kwargs['batch_size'] if 'batch_size' in kwargs else 128 118 kwargs['batch_size'] = kwargs['batch_size'] if 'batch_size' in kwargs else 128
119 kwargs['start_step'] = kwargs['start_step'] if 'start_step' in kwargs else 0 119 kwargs['start_step'] = kwargs['start_step'] if 'start_step' in kwargs else 0
120 - kwargs['max_step'] = kwargs['max_step'] if 'max_step' in kwargs else 64000 120 + kwargs['max_step'] = kwargs['max_step'] if 'max_step' in kwargs else 6500
121 kwargs['fast_auto_augment'] = kwargs['fast_auto_augment'] if 'fast_auto_augment' in kwargs else False 121 kwargs['fast_auto_augment'] = kwargs['fast_auto_augment'] if 'fast_auto_augment' in kwargs else False
122 kwargs['augment_path'] = kwargs['augment_path'] if 'augment_path' in kwargs else None 122 kwargs['augment_path'] = kwargs['augment_path'] if 'augment_path' in kwargs else None
123 123
......