4 files changed, 6 insertions, 7 deletions
diff --git a/app/client/modules/samplernn/samplernn.actions.js b/app/client/modules/samplernn/samplernn.actions.js
index e204382..e3e735f 100644
--- a/app/client/modules/samplernn/samplernn.actions.js
+++ b/app/client/modules/samplernn/samplernn.actions.js
@@ -16,7 +16,6 @@ export const load_directories = (id) => (dispatch) => {
     actions.socket.list_directory({ module: 'samplernn', dir: 'output' }),
     load_loss()(dispatch),
   ], (percent, i, n) => {
-    console.log(percent + "%")
     dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
     // console.log(res)
diff --git a/app/client/modules/samplernn/samplernn.datasets.js b/app/client/modules/samplernn/samplernn.datasets.js
index c10a1db..08f43c2 100644
--- a/app/client/modules/samplernn/samplernn.datasets.js
+++ b/app/client/modules/samplernn/samplernn.datasets.js
@@ -62,8 +62,6 @@ class SampleRNNDatasets extends Component {
   render(){
     const { samplernn, match, history } = this.props
     if (samplernn.loading) {
-      // console.log('loading')
-      console.log(samplernn.loading)
       return <Loading progress={samplernn.progress} />
     }
     if (!samplernn.data.folders.length) {
diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js
index 8b3b8e6..3962755 100644
--- a/app/relay/modules/samplernn.js
+++ b/app/relay/modules/samplernn.js
@@ -16,8 +16,8 @@ const train = {
   script: 'train.py',
   params: (task) => {
     return [
-      '--exp', task.dataset,
-      '--dataset', task.dataset,
+      '--exp', task.dataset.name,
+      '--dataset', task.dataset.name,
       '--frame_sizes', '8', '2',
       '--n_rnn', '2',
       '--epoch_limit', task.epochs || 4,
diff --git a/app/relay/runner.js b/app/relay/runner.js
index 811dff3..8231e3b 100644
--- a/app/relay/runner.js
+++ b/app/relay/runner.js
@@ -89,9 +89,10 @@ export function status () {
 
 export function build_params(module, activity, task) {
 	const interpreter = interpreters[activity.type]
-  let opt_params;
+  let opt_params, activity_params;
 	if (typeof activity.params === 'function') {
 		opt_params = activity.params(task)
+    activity_params = []
 	}
 	else {
 		const opt = task.opt || {}
@@ -103,10 +104,11 @@ export function build_params(module, activity, task) {
 			}
 			return [flag, value]
 		}).reduce((acc, cur) => acc.concat(cur), [])
+    activity_params = activity.params
 	}
   const params = (interpreter.params || [])
     .concat([ activity.script ])
-    .concat(activity.params || [])
+    .concat(activity_params)
     .concat(opt_params)
 	return {
 		activity,