1 files changed, 5 insertions, 1 deletions
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index bd51f2b..412f066 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -26,13 +26,17 @@ export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
   const task = {
     module: module.name,
     activity: 'train',
-    dataset: dataset.name,
+    dataset: dataset.name || dataset,
     epoch: 0,
     epochs: epochs,
     opt: {
       folder_id: folder_id,
     }
   }
+  if (!task.dataset) {
+    console.error("train task: no dataset specified")
+    return
+  }
   console.log(task)
   return actions.queue.add_task(task)
 }