I'll merge these commits later. Promise.

nasframe/scripts/cli.py

@@ -97,18 +97,15 @@ def find_best(filename, config, preprocess, draw, save_to):
 @click.option('--force-perprocess', help=
               'Will force preprocessing, even if preprocessed data exists.',
               is_flag=True, default=False)
-@click.option('--curriculum', help=
-              'If set, will perform curriculum architect training.',
-              is_flag=True, default=False)
-def toxic(num_gpus, val_fraction, resume, config_path, gpu_idx, force_perprocess, curriculum):
+def toxic(num_gpus, val_fraction, resume, config_path, gpu_idx, force_perprocess):
     """
     Preforms neural architecture search on Jigsaw Toxic Comment dataset
     """
     assert 0 < val_fraction < 1, 'Validation data fraction has to be in range (0,1).'
     assert num_gpus >= 1, 'Number of GPUs has to be >= 1.'
 
     from .toxiccomment import train_toxic
-    train_toxic(num_gpus, val_fraction, resume, config_path, gpu_idx, force_perprocess, curriculum)
+    train_toxic(num_gpus, val_fraction, resume, config_path, gpu_idx, force_perprocess)
 
 
 @nas.command()

nasframe/scripts/toxiccomment.py

@@ -75,7 +75,7 @@ def prepare_data(data_dir, val_fraction, embedding_path, embedding_dim, force_pe
     return embedding
 
 
-def train_toxic(num_gpus, val_fraction, resume, config_path, gpu_idx, force_perprocess, curriculum):
+def train_toxic(num_gpus, val_fraction, resume, config_path, gpu_idx, force_perprocess):
     """
     Trains architect (performs NAS) of Jigsaw Toxic Comment dataset.
     See cli help, for parameter description.
@@ -102,6 +102,7 @@ def train_toxic(num_gpus, val_fraction, resume, config_path, gpu_idx, force_perp
     )
 
     input_shape = config['child_training'].pop('input_shape')
+    curriculum = config['architect_training'].get('curriculum', False)
     if curriculum:
         storage = train_curriculum(
             config, worker_fn,