{# Copyright (c) 2014-2016, NVIDIA CORPORATION. All rights reserved. #} {% from "helper.html" import print_flashes %} {% from "helper.html" import print_errors %} {% from "helper.html" import mark_errors %} {% extends "layout.html" %} {% block title %} New Image Classification Model {% endblock %} {% block nav %}
  • New Model
  • {% endblock %} {% block content %}
    {{ form.hidden_tag() }} {{ print_errors(form) }}
    {{form.dataset.label}} {{form.dataset.tooltip}} {{form.dataset(class='form-control', size=5)}}
    {{form.python_layer_client_file.label}} {{form.python_layer_client_file.tooltip}} {{form.python_layer_client_file(class='form-control')}}
    {{form.python_layer_server_file.label}} {{form.python_layer_server_file.tooltip}} {{form.python_layer_server_file(class='form-control autocomplete_path')}}

    Data Transformations

    {{form.crop_size.label}} {{form.crop_size.tooltip}} {{form.crop_size(class='form-control', placeholder='none')}}
    {{form.use_mean.label}} {{form.use_mean.tooltip}} {{form.use_mean(class='form-control')}}

    Solver Options

    {{form.train_epochs.label}} {{form.train_epochs.tooltip}} {{form.train_epochs(class='form-control')}}
    {{form.snapshot_interval.label}} {{form.snapshot_interval.tooltip}} {{form.snapshot_interval(class='form-control')}}
    {{form.val_interval.label}} {{form.val_interval.tooltip}} {{form.val_interval(class='form-control')}}
    {# TODO: neat progress bar #}
    {{form.random_seed.label}} {{form.random_seed.tooltip}} {{form.random_seed(class='form-control', placeholder='[none]')}}
    {{form.batch_size.label}} {{form.batch_size.tooltip}} {{form.batch_size(class='form-control', placeholder='[network defaults]')}}
    {{form.solver_type.label}} {{form.solver_type.tooltip}} {{form.solver_type(class='form-control')}}
    {{form.learning_rate.label}} {{form.learning_rate.tooltip}} {{form.learning_rate(class='form-control learning-rate-option')}}

    {{form.framework(class='form-control')}}
    {% for network in form.standard_networks %} {% if network.data == 'lenet' %} {% elif network.data == 'alexnet' %} {% elif network.data == 'vgg-16' %} {% elif network.data == 'googlenet' %} {% else %} {% endif %} {% endfor %}
    Network Details Intended image size
    {{network}} {{network.label}} Original paper [1998] 28x28 (gray) Original paper [2012] 256x256 Original paper [2014] 256x256 Original paper [2014] 256x256?Customize
    {% set batch_size = 10 %} {% for batch in form.previous_networks|batch(batch_size) %} {% set batch_loop_index = loop.index0 %}
    {% for network in batch %} {% set inner_index = batch_loop_index * batch_size + loop.index0 %} {% set previous_job = previous_networks_fullinfo[inner_index] %} {% else %} {% endfor %}
    Network Pretrained Model
    {{network}} {{network.label}} View {{previous_job.train_task().get_framework_id()}} {% set snapshot_list = previous_network_snapshots[inner_index] %} {% if snapshot_list|length %} {% endif %} Customize
    None
    {% endfor %}
    {% if form.previous_networks.choices|length > batch_size %}
      {% for i in range((form.previous_networks.choices|length/batch_size)|round(0,'ceil')|int) %}
    • {{i+1}}
    • {% endfor %}
    {% endif %}

    {{form.custom_network.label}} {{form.custom_network.explanation(file='/models/images/classification/custom_network_explanation.html')}} {{form.custom_network(class='form-control', rows=10)}}
    {{form.custom_network_snapshot.label}} {{form.custom_network_snapshot.tooltip}} {{form.custom_network_snapshot(class='form-control autocomplete_path')}}
    {{ form.method(style="display:none;") }}
    {% if form.select_gpu.choices|length > 2 and not multi_gpu %}
    {{form.select_gpu.label.text}}
    {% for choice in form.select_gpu %}
    {% endfor %}
    {% endif %} {% if form.select_gpus.choices| length > 1 and multi_gpu %}
    {{form.select_gpu_count.label}} {{form.select_gpu_count(class='form-control')}}

    or

    {{form.select_gpus.label}} {{form.select_gpus.tooltip}} {{form.select_gpus(class='form-control', size=4)}}
    {% endif %}
    {{form.model_name.label}} {{form.model_name.tooltip}} {{form.model_name(class='form-control')}}
    {% endblock %}