diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-08 00:34:05 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-08 00:34:05 +0200 |
| commit | fec93601f163d22760978c9db61c4a78ce5281da (patch) | |
| tree | d9bfe66d6e144f478997a7bda9610fc73f3c440a /app | |
| parent | 8850e41b3c51a5d90ca6c1667d6c1b1cebae6ad0 (diff) | |
samplernn tasks
Diffstat (limited to 'app')
| -rw-r--r-- | app/client/modules/pix2pix/views/pix2pix.live.js | 4 | ||||
| -rw-r--r-- | app/client/modules/samplernn/samplernn.tasks.js | 1 | ||||
| -rw-r--r-- | app/relay/modules/samplernn.js | 12 |
3 files changed, 17 insertions, 0 deletions
diff --git a/app/client/modules/pix2pix/views/pix2pix.live.js b/app/client/modules/pix2pix/views/pix2pix.live.js index 0457668..a6e739c 100644 --- a/app/client/modules/pix2pix/views/pix2pix.live.js +++ b/app/client/modules/pix2pix/views/pix2pix.live.js @@ -118,6 +118,10 @@ class Pix2PixLive extends Component { value={(this.props.frame.sequence_i || 0) / (this.props.frame.sequence_len || 1)} onChange={this.seek} /> + <Slider live + name='frame_delay' + min={0.1} max={5.0} type='float' + /> {this.renderRestartButton()} <Button title={ diff --git a/app/client/modules/samplernn/samplernn.tasks.js b/app/client/modules/samplernn/samplernn.tasks.js index d527254..f253047 100644 --- a/app/client/modules/samplernn/samplernn.tasks.js +++ b/app/client/modules/samplernn/samplernn.tasks.js @@ -10,6 +10,7 @@ export const train_task = (dataset, folder_id, epochs=1) => dispatch => { module: 'samplernn', activity: 'train', dataset: dataset.name, + epoch: dataset.checkpoints.length ? dataset.checkpoints[0].epoch || 0 : 0 epochs: epochs, opt: { folder_id: folder_id, diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js index 4ce4f23..668ab24 100644 --- a/app/relay/modules/samplernn.js +++ b/app/relay/modules/samplernn.js @@ -37,6 +37,18 @@ const train = { '--keep_old_checkpoints', task.opt.keep_old_checkpoints ? 'True' : 'False', ] }, + listen: (task, line, i) => { + // ################################################################################ + // Epoch summary: + // training_loss: 1.4509 validation_loss: 1.4260 test_loss: 1.4316 + // ################################################################################ + // + if ( line.match(/Epoch summary/) ) { + task.epoch += 1 + return { type: 'progress', action: 'epoch', task, } + } + return null + } after: publish, } const generate = { |