diff options
Diffstat (limited to 'app/client')
| -rw-r--r-- | app/client/modules/pix2pixhd/views/pix2pixhd.train.js | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js index b00947f..20f6fc8 100644 --- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js +++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js @@ -101,7 +101,10 @@ class Pix2PixHDTrain extends Component { if (folder.name === 'results') return const datasets = folder.datasets.map(name => { const dataset = datasetLookup[name] - const file = this.props.pix2pixhd.data.fileLookup[id] + let file; + if (dataset.input.length) { + file = this.props.pix2pixhd.data.fileLookup[dataset.input[0]] + } // console.log(folder) if (dataset.isBuilt) { // && dataset.checkpoints.length if (file) { |