From 57726902fe70f6c051c20f7104a9b78687c15479 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 4 Jun 2018 03:52:59 +0200
Subject: progress barz

---
 app/client/modules/samplernn/samplernn.actions.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'app/client/modules/samplernn/samplernn.actions.js')

diff --git a/app/client/modules/samplernn/samplernn.actions.js b/app/client/modules/samplernn/samplernn.actions.js
index 343550e..e204382 100644
--- a/app/client/modules/samplernn/samplernn.actions.js
+++ b/app/client/modules/samplernn/samplernn.actions.js
@@ -17,6 +17,7 @@ export const load_directories = (id) => (dispatch) => {
     load_loss()(dispatch),
   ], (percent, i, n) => {
     console.log(percent + "%")
+    dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
     // console.log(res)
     const [folders, files, tasks, datasets, results, output, lossReport] = res
@@ -262,12 +263,12 @@ export const fetch_url = (url) => (dispatch) => {
   }, { preempt: true, watch: true })
 }
 
-export const train_task_now = (dataset) => (dispatch) => {
+export const train_task_now = (dataset, epochs=1) => (dispatch) => {
   const task = {
     module: 'samplernn',
     activity: 'train',
     dataset: dataset,
-    epochs: 6,
+    epochs: epochs,
     opt: {
       sample_length: 44100 * 5,
       n_samples: 6,
-- 
cgit v1.2.3-70-g09d2