more intuitive train function

KevinMenden · Dec 21, 2020 · fe12c05 · fe12c05
1 parent 0e6fe62
commit fe12c05
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/scaden/__main__.py b/scaden/__main__.py
@@ -56,9 +56,11 @@ def cli():
                 type=click.Path(exists=True),
                 required=True,
                 metavar='<training data>')
-@click.option('--train_datasets',
-              default='',
-              help='Datasets used for training. Uses all by default.')
+@click.option(
+    '--train_datasets',
+    default='',
+    help=
+    'Comma-separated list of datasets used for training. Uses all by default.')
 @click.option('--model_dir', default="./", help='Path to store the model in')
 @click.option('--batch_size',
               default=128,

diff --git a/scaden/train.py b/scaden/train.py
@@ -48,7 +48,7 @@ def training(data_path,
     if train_datasets == '':
         train_datasets = []
     else:
-        train_datasets = train_datasets.split()
+        train_datasets = train_datasets.split(',')
         print(f"Training on: {train_datasets}")
 
     # Training of M256 model