Previous network clone epoch

.travis.yml

@@ -121,6 +121,7 @@ before_install:
     - deactivate
     - virtualenv --system-site-packages ~/venv
     - source ~/venv/bin/activate
+    - sudo apt-get install libboost-filesystem1.55-dev libboost-python1.55-dev libboost-system1.55-dev libboost-thread1.55-dev
 
 install:
     - mkdir -p ~/.config/matplotlib

digits/model/forms.py

@@ -284,6 +284,12 @@ def validate_lr_multistep_values(form, field):
         ],
     )
 
+    # This is used to save and restore for cloning because the pretrained model selections are
+    # created on the fly in the html file. This form is not displayed.
+    saved_epoch = utils.forms.TextField(
+        'Saved epoch',
+    )
+
     custom_network = utils.forms.TextAreaField(
         'Custom Network',
         validators=[

digits/model/images/classification/views.py

@@ -204,8 +204,10 @@ def create():
                                     "Pretrained_model for the selected epoch doesn't exist. "
                                     "May be deleted by another user/process. "
                                     "Please restart the server to load the correct pretrained_model details.")
-                        break
 
+                            # terrible hack to save the epoch for cloning
+                            form.saved_epoch.data = epoch
+                        break
             elif form.method.data == 'pretrained':
                 pretrained_job = scheduler.get_job(form.pretrained_networks.data)
                 model_def_path = pretrained_job.get_model_def_path()

digits/model/images/generic/views.py

@@ -167,6 +167,9 @@ def create(extension_id=None):
                                     "Pretrained_model for the selected epoch doesn't exist. "
                                     "May be deleted by another user/process. "
                                     "Please restart the server to load the correct pretrained_model details.")
+
+                            # terrible hack to save the epoch for cloning
+                            form.saved_epoch.data = epoch
                         break
             elif form.method.data == 'pretrained':
                 pretrained_job = scheduler.get_job(form.pretrained_networks.data)

digits/templates/models/images/classification/partials/new/network_tab_previous.html

@@ -26,7 +26,7 @@
                     {% if snapshot_list|length %}
                     <select class="form-control" id="{{network.data}}-snapshot" name="{{network.data}}-snapshot">
                         {% for each_epoch in snapshot_list %}
-                        <option value="{{each_epoch[0]}}">{{each_epoch[1]}}</option>
+                        <option value="{{each_epoch[0]}}" {{'selected' if each_epoch[0] == form.saved_epoch.data}}>{{each_epoch[1]}}</option>
                         {% endfor %}
                     </select>
                     {% endif %}
@@ -39,7 +39,6 @@
             </tr>
             {% endfor %}
         </table>
-
     </div>
     {% endfor %}
 </div>

digits/templates/models/images/generic/partials/new/network_tab_previous.html

@@ -25,7 +25,7 @@
                     {% if snapshot_list|length %}
                     <select class="form-control" id="{{network.data}}-snapshot" name="{{network.data}}-snapshot">
                         {% for each_epoch in snapshot_list %}
-                        <option value="{{each_epoch[0]}}">{{each_epoch[1]}}</option>
+                        <option value="{{each_epoch[0]}}" {{'selected' if each_epoch[0] == form.saved_epoch.data}}>{{each_epoch[1]}}</option>
                         {% endfor %}
                     </select>
                     {% endif %}