diff options
Diffstat (limited to 'app/relay/modules')
| -rw-r--r-- | app/relay/modules/samplernn.js | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js index 8b3b8e6..3962755 100644 --- a/app/relay/modules/samplernn.js +++ b/app/relay/modules/samplernn.js @@ -16,8 +16,8 @@ const train = { script: 'train.py', params: (task) => { return [ - '--exp', task.dataset, - '--dataset', task.dataset, + '--exp', task.dataset.name, + '--dataset', task.dataset.name, '--frame_sizes', '8', '2', '--n_rnn', '2', '--epoch_limit', task.epochs || 4, |