1 files changed, 5 insertions, 0 deletions
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
index e5bd7f6..b00947f 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
@@ -101,7 +101,12 @@ 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]
+        // console.log(folder)
         if (dataset.isBuilt) { // && dataset.checkpoints.length
+          if (file) {
+            return file.name
+          }
           return name
         }
         return null