From fec93601f163d22760978c9db61c4a78ce5281da Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 8 Jun 2018 00:34:05 +0200 Subject: samplernn tasks --- app/relay/modules/samplernn.js | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'app/relay/modules/samplernn.js') 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 = { -- cgit v1.2.3-70-g09d2