1 files changed, 3 insertions, 2 deletions
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,