diff options
Diffstat (limited to 'app/client')
| -rw-r--r-- | app/client/dataset/dataset.loader.js | 1 | ||||
| -rw-r--r-- | app/client/modules/pix2pixhd/views/pix2pixhd.train.js | 18 |
2 files changed, 18 insertions, 1 deletions
diff --git a/app/client/dataset/dataset.loader.js b/app/client/dataset/dataset.loader.js index 76d8b16..f1ff75e 100644 --- a/app/client/dataset/dataset.loader.js +++ b/app/client/dataset/dataset.loader.js @@ -20,6 +20,7 @@ export const unsortedFolder = (module, instantiate=false) => { export const emptyDataset = (module, datasetLookup, name, folder) => { const dataset = { name, + folder_id: folder.id, input: [], checkpoints: [], output: [], diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js index 78def7a..d4e390c 100644 --- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js +++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js @@ -152,8 +152,24 @@ class Pix2PixHDTrain extends Component { renderEditor(){ const { pix2pixhd, queue, remote } = this.props - const { checkpoint, folder_id } = pix2pixhd const { dataset, processed } = this.state + let { checkpoint, folder_id } = pix2pixhd + + const currentDataset = this.props.pix2pixhd.data.datasetLookup[this.state.dataset] + if (currentDataset && currentDataset.folder_id) { + folder_id = currentDataset.folder_id || folder_id + console.log("dataset folder id", folder_id) + const processed = currentDataset.input + .map(f => this.props.pix2pixhd.data.fileLookup[f]) + .reduce((a, b) => { + console.log(b) + return b.processed || a + }, false) + console.log("is processed?", processed) + } + console.log(currentDataset) + console.log("so folder id is..", folder_id) + return ( <div> <Group title='Sequence Editor'> |