From 198c41778cfe39d37aab92c08325dc8c9d4437a0 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 3 Jun 2018 03:48:22 +0200 Subject: condense samplernn tasks --- app/relay/modules/samplernn.js | 34 ++++++++++++++++++++-------------- 1 file changed, 20 insertions(+), 14 deletions(-) (limited to 'app/relay/modules/samplernn.js') diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js index 08550ed..3cef25d 100644 --- a/app/relay/modules/samplernn.js +++ b/app/relay/modules/samplernn.js @@ -15,15 +15,18 @@ const train = { type: 'pytorch', script: 'train.py', params: (task) => { + return [ + '--exp', task.checkpoint, + '--dataset', task.dataset, + '--frame_sizes', '8', '2', + '--n_rnn', '2', + '--epoch_limit', task.opt.epoch_limit, + '--sample_length', task.opt.sample_length, + '--n_samples', task.opt.n_samples, + '--keep_old_checkpoints', task.opt.keep_old_checkpoints ? 'True' : 'False', + ] }, onComplete: publish, - // python train.py \ - // --exp $checkpoint_name --dataset $dataset_name \ - // --frame_sizes 8 2 --n_rnn 2 \ - // --sample_length $sample_length \ - // --n_samples $n_samples \ - // --keep_old_checkpoints False \ - // --epoch_limit $epoch_limit \ } const report = { type: 'perl', @@ -35,20 +38,23 @@ const generate = { type: 'pytorch', script: 'generate.py', params: (task) => { + return [ + '--exp', task.checkpoint, + '--dataset', task.dataset, + '--frame_sizes', '8', '2', + '--n_rnn', '2', + '--sample_length', task.opt.sample_length, + '--n_samples', task.opt.n_samples, + '--keep_old_checkpoints', task.opt.keep_old_checkpoints ? 'True' : 'False', + ] }, onComplete: publish, - // python generate.py \ - // --exp $checkpoint_name --dataset $dataset_name \ - // --frame_sizes 8 2 --n_rnn 2 \ - // --sample_length $sample_length \ - // --n_samples $n_samples \ - // --keep_old_checkpoints False \ - // --epoch_limit $epoch_limit \ } const publish = { type: 'perl', script: 'latest.pl', params: (task) => { + return ['-l', task.dataset] } } // after train and generate, run perl latest.pl -l $checkpoint_name -- cgit v1.2.3-70-g09d2