diff options
Diffstat (limited to 'app/relay/modules/samplernn.js')
| -rw-r--r-- | app/relay/modules/samplernn.js | 34 |
1 files changed, 20 insertions, 14 deletions
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 |