From aa09790d92deca05d86b243072ed7bca697e5469 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Sun, 7 Oct 2018 04:14:56 +0200
Subject: request latest

---
 app/client/modules/pix2pixhd/views/pix2pixhd.train.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'app/client/modules/pix2pixhd')

diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
index 1a23618..78def7a 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
@@ -101,7 +101,7 @@ class Pix2PixHDTrain extends Component {
       if (folder.name === 'results') return
       const datasets = folder.datasets.map(name => {
         const dataset = datasetLookup[name]
-        if (dataset.isBuilt && dataset.checkpoints.length) {
+        if (dataset.isBuilt) { // && dataset.checkpoints.length
           return name
         }
         return null
@@ -112,6 +112,8 @@ class Pix2PixHDTrain extends Component {
       }
     }).filter(n => !!n && !!n.options.length).sort((a,b) => a.name.localeCompare(b.name))
 
+    let epochs = this.props.pix2pixhd.epochs
+    if (!epochs || !epochs.length) epochs = ["latest"]
     // console.log('state', this.props.pix2pixhd.data.epochs)
     // console.log(this.state.dataset, this.state.epoch)
     // console.log(queue)
@@ -134,7 +136,7 @@ class Pix2PixHDTrain extends Component {
               <Select
                 title="Epoch"
                 name="epoch"
-                options={this.props.pix2pixhd.epochs}
+                options={epochs}
                 onChange={this.handleChange}
                 value={this.state.epoch}
               />
-- 
cgit v1.2.3-70-g09d2