From d0b5aee1fbd50d58c418952cd649c6ed28260e23 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 23 Sep 2018 20:37:43 +0200 Subject: more tasklist stuff --- app/client/modules/pix2pixhd/pix2pixhd.tasks.js | 20 +++++++++++++------- 1 file changed, 13 insertions(+), 7 deletions(-) (limited to 'app/client/modules/pix2pixhd/pix2pixhd.tasks.js') diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js index 8fec652..89184d2 100644 --- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js +++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js @@ -23,10 +23,12 @@ export const fetch_task = (url, file_id, dataset) => dispatch => { } export const train_task = (dataset, folder_id, epochs=1) => dispatch => { + dataset = dataset.name || dataset + if (dataset === 'PLACEHOLDER') return const task = { module: module.name, activity: 'train', - dataset: dataset.name || dataset, + dataset, epoch: 0, epochs: epochs, opt: { @@ -58,6 +60,7 @@ export const live_task = (sequence, checkpoint, opt) => dispatch => { } export const augment_task = (dataset, opt) => dispatch => { + if (dataset === 'PLACEHOLDER') return const task = { module: module.name, activity: 'augment', @@ -72,14 +75,17 @@ export const augment_task = (dataset, opt) => dispatch => { return actions.queue.add_task(task) } -export const clear_recursive_task = (dataset) => dispatch => { +export const create_dataset_task = (opt) => dispatch => { const task = { module: module.name, - activity: 'clear_recursive', - dataset, + activity: 'create_dataset', + dataset: opt.title, + folder_id: opt.folder_id, + opt: { + ...opt, + } } console.log(task) - console.log('add clear recursive task') + console.log('add create_dataset task') return actions.queue.add_task(task) -} - +} \ No newline at end of file -- cgit v1.2.3-70-g09d2