diff options
Diffstat (limited to 'app/client/modules/pix2pixhd/pix2pixhd.tasks.js')
| -rw-r--r-- | app/client/modules/pix2pixhd/pix2pixhd.tasks.js | 20 |
1 files changed, 13 insertions, 7 deletions
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 |
