1 files changed, 12 insertions, 0 deletions
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 = {