fix module

3 files changed, 21 insertions, 12 deletions

diff --git a/app/client/common/select.component.js b/app/client/common/select.component.js
index f1e441f..fc69e87 100644
--- a/app/client/common/select.component.js
+++ b/app/client/common/select.component.js
@@ -13,21 +13,29 @@ class Select extends Component {
     this.props.onChange && this.props.onChange(this.props.name, new_value)
   }
   render() {
-    const value = this.props.opt[this.props.name]
+    const currentValue = this.props.opt[this.props.name]
     let lastValue
     const options = (this.props.options || []).map((key,i) => {
       let name, value
-      if (typeof key === 'object' && key.length) {
-        [name, value] = key
-      }
-      else if (typeof key === 'string') {
+      if (typeof key === 'string') {
         name = key.length < 4 ? key.toUpperCase() : key
         value = key
-      } else {
-        let frames = Math.round(key.count / 30) + ' s.'
-        name = key.name.replace(/_/g, ' ') + ' (' + frames + ')'
-        value = key.name
       }
+      else if (typeof key === 'object') {
+        if (key.length) {
+          [name, value] = key
+        }
+        else if (key.count) {
+          let frames = Math.round(key.count / 30) + ' s.'
+          name = key.name.replace(/_/g, ' ') + ' (' + frames + ')'
+          value = key.name
+        }
+        else {
+          name = key.name
+          value = key.value || key.name
+        }
+      }
+      console.log(key, name, value)
       lastValue = value
       return (
         <option value={value} key={i}>
@@ -41,7 +49,7 @@ class Select extends Component {
           <span>{this.props.title}</span>
           <select
             onChange={this.handleChange}
-            value={value || lastValue}
+            value={currentValue || lastValue}
           >
             {options}
           </select>
diff --git a/app/client/modules/pix2wav/views/pix2wav.live.js b/app/client/modules/pix2wav/views/pix2wav.live.js
index 9a61e60..256b6aa 100644
--- a/app/client/modules/pix2wav/views/pix2wav.live.js
+++ b/app/client/modules/pix2wav/views/pix2wav.live.js
@@ -97,13 +97,13 @@ class Pix2WavLive extends Component {
               <Select
                 name='sequence_name'
                 title='sequence'
-                options={this.props.pix2wav.data.sequences.map(file => file.name)}
+                options={this.props.pix2wav.data.sequences}
                 onChange={this.changeSequence}
               />
               <Select
                 name='checkpoint_name'
                 title='checkpoint'
-                options={this.props.pix2wav.data.checkpoints.map(file => file.name)}
+                options={this.props.pix2wav.data.checkpoints}
                 onChange={this.changeCheckpoint}
               />
               <Select
diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js
index 848c859..5dbc847 100644
--- a/app/relay/modules/pix2pix.js
+++ b/app/relay/modules/pix2pix.js
@@ -94,6 +94,7 @@ const live = {
     return [
       '--dataroot', path.join(cwd, 'sequences', task.module, task.dataset),
       '--start_img', path.join(cwd, 'sequences', task.module, task.dataset, 'frame_00001.png'),
+      '--checkpoint-name', task.checkpoint,
       '--experiment', task.checkpoint,
       '--name', task.checkpoint,
       '--module-name', task.module,