Merge branch 'master' into upgradeTFver

vanvalenlab · Aug 6, 2019 · df36d04 · df36d04
2 parents 9b80e4b + f3b1024
commit df36d04
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/deepcell/notebooks/train.py b/deepcell/notebooks/train.py
@@ -321,13 +321,12 @@ def make_notebook(data,
     # Set up training parameters
     training_kwargs = {
         'model': 'model',
-        'dataset': 'DATA_FILE',
+        'dataset': 'os.path.join(DATA_DIR, DATA_FILE + ".npz")',
         'expt': '"{}"'.format(train_type + ('_' + transform if transform else '')),
         'optimizer': 'optimizer',
         'batch_size': 'batch_size',
         'n_epoch': 'n_epoch',
-        'direc_save': 'MODEL_DIR',
-        'direc_data': 'NPZ_DIR',
+        'model_dir': 'MODEL_DIR',
         'log_dir': 'LOG_DIR',
         'lr_sched': 'lr_sched',
         'rotation_range': 180,