diff options
Diffstat (limited to 'app')
| -rw-r--r-- | app/client/modules/samplernn/datasets.component.js | 2 | ||||
| -rw-r--r-- | app/relay/modules/samplernn.js | 8 |
2 files changed, 5 insertions, 5 deletions
diff --git a/app/client/modules/samplernn/datasets.component.js b/app/client/modules/samplernn/datasets.component.js index 63d3ab4..b6df614 100644 --- a/app/client/modules/samplernn/datasets.component.js +++ b/app/client/modules/samplernn/datasets.component.js @@ -84,7 +84,7 @@ class SampleRNNDatasets extends Component { fetchURL(url) { console.log(url) this.props.actions.task.start_task({ - activity: 'cpu', + activity: 'fetch', module: 'samplernn', dataset: 'test', epochs: 1, diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js index 01a3b82..601acbb 100644 --- a/app/relay/modules/samplernn.js +++ b/app/relay/modules/samplernn.js @@ -6,7 +6,7 @@ const cwd = process.env.SAMPLERNN_CWD || path.join(process.env.HOME, 'code/' + n const fetch = { type: 'perl', script: 'get.pl', - params: (task) => { + build_params: (task) => { console.log(task) return [ task.opt.url ] } @@ -14,7 +14,7 @@ const fetch = { const train = { type: 'pytorch', script: 'train.py', - params: (task) => { + build_params: (task) => { }, onComplete: publish, // python train.py \ @@ -28,7 +28,7 @@ const train = { const generate = { type: 'pytorch', script: 'generate.py', - params: (task) => { + build_params: (task) => { }, onComplete: publish, // python generate.py \ @@ -42,7 +42,7 @@ const generate = { const publish = { type: 'perl', script: 'latest.pl', - params: (task) => { + build_params: (task) => { } } // after train and generate, run perl latest.pl -l $checkpoint_name |