blob: 65781e4356b0b66854ffa72c59cb2c1d9a1215b2 [file] [log] [blame]
<!DOCTYPE html>
<!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]-->
<!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]-->
<head>
<meta charset="utf-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Quickstart - Cifar10 example &mdash; singa 2.0.0 documentation</title>
<script type="text/javascript" src="../_static/js/modernizr.min.js"></script>
<script type="text/javascript">
var DOCUMENTATION_OPTIONS = {
URL_ROOT:'../',
VERSION:'2.0.0',
LANGUAGE:'None',
COLLAPSE_INDEX:false,
FILE_SUFFIX:'.html',
HAS_SOURCE: true,
SOURCELINK_SUFFIX: '.txt'
};
</script>
<script type="text/javascript" src="../_static/jquery.js"></script>
<script type="text/javascript" src="../_static/underscore.js"></script>
<script type="text/javascript" src="../_static/doctools.js"></script>
<script type="text/javascript" src="../_static/js/theme.js"></script>
<link rel="stylesheet" href="../_static/css/theme.css" type="text/css" />
<link rel="stylesheet" href="../_static/pygments.css" type="text/css" />
<link rel="index" title="Index" href="../genindex.html" />
<link rel="search" title="Search" href="../search.html" />
<link href="../_static/style.css" rel="stylesheet" type="text/css">
<!--link href="../_static/fontawesome-all.min.css" rel="stylesheet" type="text/css"-->
<link rel="stylesheet" href="https://use.fontawesome.com/releases/v5.0.13/css/all.css"
integrity="sha384-DNOHZ68U8hZfKXOrtjWvjxusGo9WQnrNx2sqG0tfsghAvtVlRW3tvkXWZh58N9jp" crossorigin="anonymous">
<style>
.fa:hover {
opacity: 0.7;
}
.fab:hover {
opacity: 0.7;
}
</style>
</head>
<body class="wy-body-for-nav">
<div class="wy-grid-for-nav">
<nav data-toggle="wy-nav-shift" class="wy-nav-side">
<div class="wy-side-scroll">
<div class="wy-side-nav-search" >
<a href="../index.html" class="icon icon-home"> singa
<img src="../_static/singa.png" class="logo" alt="Logo"/>
</a>
<div class="version">
latest
</div>
<div role="search">
<form id="rtd-search-form" class="wy-form" action="../search.html" method="get">
<input type="text" name="q" placeholder="Search docs" />
<input type="hidden" name="check_keywords" value="yes" />
<input type="hidden" name="area" value="default" />
</form>
</div>
</div>
<div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="main navigation">
<ul>
<li class="toctree-l1"><a class="reference internal" href="index.html">Documentation</a></li>
<li class="toctree-l1"><a class="reference internal" href="../downloads.html">Download SINGA</a></li>
<li class="toctree-l1"><a class="reference internal" href="../security.html">Security</a></li>
</ul>
<p class="caption"><span class="caption-text">Development</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../develop/contribute-code.html">How to Contribute Code</a></li>
<li class="toctree-l1"><a class="reference internal" href="../develop/contribute-docs.html">How to Contribute to Documentation</a></li>
<li class="toctree-l1"><a class="reference internal" href="../develop/how-to-release.html">How to prepare a release</a></li>
</ul>
<p class="caption"><span class="caption-text">Community</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../community/source-repository.html">Source Repository</a></li>
<li class="toctree-l1"><a class="reference internal" href="../community/mail-lists.html">Project Mailing Lists</a></li>
<li class="toctree-l1"><a class="reference internal" href="../community/issue-tracking.html">Issue Tracking</a></li>
<li class="toctree-l1"><a class="reference internal" href="../community/team-list.html">The SINGA Team</a></li>
</ul>
</div>
</div>
</nav>
<section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
<nav class="wy-nav-top" aria-label="top navigation">
<i data-toggle="wy-nav-top" class="fa fa-bars"></i>
<a href="../index.html">singa</a>
</nav>
<div class="wy-nav-content">
<div class="rst-content">
<div role="navigation" aria-label="breadcrumbs navigation">
<ul class="wy-breadcrumbs">
<li><a href="../index.html">Docs</a> &raquo;</li>
<li>Quickstart - Cifar10 example</li>
<li class="wy-breadcrumbs-aside">
</li>
</ul>
<hr/>
</div>
<div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
<div itemprop="articleBody">
<!--
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
to you under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance
with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing,
software distributed under the License is distributed on an
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
KIND, either express or implied. See the License for the
specific language governing permissions and limitations
under the License.
--><div class="section" id="quickstart-cifar10-example">
<h1>Quickstart - Cifar10 example<a class="headerlink" href="#quickstart-cifar10-example" title="Permalink to this headline"></a></h1>
<p>Convolution neural network (CNN) is a type of feed-forward artificial neural network widely used for image classification. In this example, we will use a deep CNN model to do image classification for the <a class="reference external" href="http://www.cs.toronto.edu/%7Ekriz/cifar.html">CIFAR10 dataset</a>.</p>
<div class="section" id="running-instructions-for-cpp-version">
<h2>Running instructions for CPP version<a class="headerlink" href="#running-instructions-for-cpp-version" title="Permalink to this headline"></a></h2>
<p>Please refer to <a class="reference external" href="installation.html">Installation</a> page for how to install SINGA. Currently, we CNN requires CUDNN, hence both CUDA and CUDNN should be installed and SINGA should be compiled with CUDA and CUDNN.</p>
<p>The Cifar10 dataset could be downloaded by running</p>
<div class="highlight-default"><div class="highlight"><pre><span></span># switch to cifar10 directory
$ cd ../examples/cifar10
# download data for CPP version
$ python download_data.py bin
</pre></div>
</div>
<p>&#8216;bin&#8216; is for downloading binary version of Cifar10 data.</p>
<p>During downloading, you should see the detailed output like</p>
<div class="highlight-default"><div class="highlight"><pre><span></span> Downloading CIFAR10 from http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz
The tar file does exist. Extracting it now..
Finished!
</pre></div>
</div>
<p>Now you have prepared the data for this Cifar10 example, the final step is to execute the <code class="docutils literal"><span class="pre">run.sh</span></code> script,</p>
<div class="highlight-default"><div class="highlight"><pre><span></span># in SINGA_ROOT/examples/cifar10/
$ ./run.sh
</pre></div>
</div>
<p>You should see the detailed output as follows: first read the data files in order, show the statistics of training and testing data, then show the details of neural net structure with some parameter information, finally illustrate the performance details during training and validation process. The number of epochs can be specified in <code class="docutils literal"><span class="pre">run.sh</span></code> file.</p>
<div class="highlight-default"><div class="highlight"><pre><span></span><span class="n">Start</span> <span class="n">training</span>
<span class="n">Reading</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="nb">bin</span><span class="o">/</span><span class="n">data_batch_1</span><span class="o">.</span><span class="n">bin</span>
<span class="n">Reading</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="nb">bin</span><span class="o">/</span><span class="n">data_batch_2</span><span class="o">.</span><span class="n">bin</span>
<span class="n">Reading</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="nb">bin</span><span class="o">/</span><span class="n">data_batch_3</span><span class="o">.</span><span class="n">bin</span>
<span class="n">Reading</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="nb">bin</span><span class="o">/</span><span class="n">data_batch_4</span><span class="o">.</span><span class="n">bin</span>
<span class="n">Reading</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="nb">bin</span><span class="o">/</span><span class="n">data_batch_5</span><span class="o">.</span><span class="n">bin</span>
<span class="n">Reading</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="nb">bin</span><span class="o">/</span><span class="n">test_batch</span><span class="o">.</span><span class="n">bin</span>
<span class="n">Training</span> <span class="n">samples</span> <span class="o">=</span> <span class="mi">50000</span><span class="p">,</span> <span class="n">Test</span> <span class="n">samples</span> <span class="o">=</span> <span class="mi">10000</span>
<span class="n">conv1</span><span class="p">(</span><span class="mi">32</span><span class="p">,</span> <span class="mi">32</span><span class="p">,</span> <span class="mi">32</span><span class="p">,</span> <span class="p">)</span>
<span class="n">pool1</span><span class="p">(</span><span class="mi">32</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="p">)</span>
<span class="n">relu1</span><span class="p">(</span><span class="mi">32</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="p">)</span>
<span class="n">lrn1</span><span class="p">(</span><span class="mi">32</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="p">)</span>
<span class="n">conv2</span><span class="p">(</span><span class="mi">32</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="p">)</span>
<span class="n">relu2</span><span class="p">(</span><span class="mi">32</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="p">)</span>
<span class="n">pool2</span><span class="p">(</span><span class="mi">32</span><span class="p">,</span> <span class="mi">8</span><span class="p">,</span> <span class="mi">8</span><span class="p">,</span> <span class="p">)</span>
<span class="n">lrn2</span><span class="p">(</span><span class="mi">32</span><span class="p">,</span> <span class="mi">8</span><span class="p">,</span> <span class="mi">8</span><span class="p">,</span> <span class="p">)</span>
<span class="n">conv3</span><span class="p">(</span><span class="mi">64</span><span class="p">,</span> <span class="mi">8</span><span class="p">,</span> <span class="mi">8</span><span class="p">,</span> <span class="p">)</span>
<span class="n">relu3</span><span class="p">(</span><span class="mi">64</span><span class="p">,</span> <span class="mi">8</span><span class="p">,</span> <span class="mi">8</span><span class="p">,</span> <span class="p">)</span>
<span class="n">pool3</span><span class="p">(</span><span class="mi">64</span><span class="p">,</span> <span class="mi">4</span><span class="p">,</span> <span class="mi">4</span><span class="p">,</span> <span class="p">)</span>
<span class="n">flat</span><span class="p">(</span><span class="mi">1024</span><span class="p">,</span> <span class="p">)</span>
<span class="n">ip</span><span class="p">(</span><span class="mi">10</span><span class="p">,</span> <span class="p">)</span>
<span class="n">conv1_weight</span> <span class="p">:</span> <span class="mf">8.09309e-05</span>
<span class="n">conv1_bias</span> <span class="p">:</span> <span class="mi">0</span>
<span class="n">conv2_weight</span> <span class="p">:</span> <span class="mf">0.00797731</span>
<span class="n">conv2_bias</span> <span class="p">:</span> <span class="mi">0</span>
<span class="n">conv3_weight</span> <span class="p">:</span> <span class="mf">0.00795888</span>
<span class="n">conv3_bias</span> <span class="p">:</span> <span class="mi">0</span>
<span class="n">ip_weight</span> <span class="p">:</span> <span class="mf">0.00798683</span>
<span class="n">ip_bias</span> <span class="p">:</span> <span class="mi">0</span>
<span class="n">Messages</span> <span class="n">will</span> <span class="n">be</span> <span class="n">appended</span> <span class="n">to</span> <span class="n">an</span> <span class="n">existed</span> <span class="n">file</span><span class="p">:</span> <span class="n">train_perf</span>
<span class="n">Messages</span> <span class="n">will</span> <span class="n">be</span> <span class="n">appended</span> <span class="n">to</span> <span class="n">an</span> <span class="n">existed</span> <span class="n">file</span><span class="p">:</span> <span class="n">val_perf</span>
<span class="n">Epoch</span> <span class="mi">0</span><span class="p">,</span> <span class="n">training</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.828369</span><span class="p">,</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.329420</span><span class="p">,</span> <span class="n">lr</span> <span class="o">=</span> <span class="mf">0.001000</span>
<span class="n">Epoch</span> <span class="mi">0</span><span class="p">,</span> <span class="n">val</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.561823</span><span class="p">,</span> <span class="n">metric</span> <span class="o">=</span> <span class="mf">0.420600</span>
<span class="n">Epoch</span> <span class="mi">1</span><span class="p">,</span> <span class="n">training</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.465898</span><span class="p">,</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.469940</span><span class="p">,</span> <span class="n">lr</span> <span class="o">=</span> <span class="mf">0.001000</span>
<span class="n">Epoch</span> <span class="mi">1</span><span class="p">,</span> <span class="n">val</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.361778</span><span class="p">,</span> <span class="n">metric</span> <span class="o">=</span> <span class="mf">0.513300</span>
<span class="n">Epoch</span> <span class="mi">2</span><span class="p">,</span> <span class="n">training</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.320708</span><span class="p">,</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.529000</span><span class="p">,</span> <span class="n">lr</span> <span class="o">=</span> <span class="mf">0.001000</span>
<span class="n">Epoch</span> <span class="mi">2</span><span class="p">,</span> <span class="n">val</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.242080</span><span class="p">,</span> <span class="n">metric</span> <span class="o">=</span> <span class="mf">0.549100</span>
<span class="n">Epoch</span> <span class="mi">3</span><span class="p">,</span> <span class="n">training</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.213776</span><span class="p">,</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.571620</span><span class="p">,</span> <span class="n">lr</span> <span class="o">=</span> <span class="mf">0.001000</span>
<span class="n">Epoch</span> <span class="mi">3</span><span class="p">,</span> <span class="n">val</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.175346</span><span class="p">,</span> <span class="n">metric</span> <span class="o">=</span> <span class="mf">0.582000</span>
</pre></div>
</div>
<p>The training details are stored in <code class="docutils literal"><span class="pre">train_perf</span></code> file in the same directory and the validation details in <code class="docutils literal"><span class="pre">val_perf</span></code> file.</p>
</div>
<div class="section" id="running-instructions-for-python-version">
<h2>Running instructions for Python version<a class="headerlink" href="#running-instructions-for-python-version" title="Permalink to this headline"></a></h2>
<p>To run CNN example in Python version, we need to compile SINGA with Python binding,</p>
<div class="highlight-default"><div class="highlight"><pre><span></span>$ mkdir build &amp;&amp; cd build
$ cmake -DUSE_PYTHON=ON ..
$ make
</pre></div>
</div>
<p>Now download the Cifar10 dataset,</p>
<div class="highlight-default"><div class="highlight"><pre><span></span># switch to cifar10 directory
$ cd ../examples/cifar10
# download data for Python version
$ python download_data.py py
</pre></div>
</div>
<p>During downloading, you should see the detailed output like</p>
<div class="highlight-default"><div class="highlight"><pre><span></span> Downloading CIFAR10 from http://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
The tar file does exist. Extracting it now..
Finished!
</pre></div>
</div>
<p>Then execute the <code class="docutils literal"><span class="pre">train.py</span></code> script to build the model</p>
<div class="highlight-default"><div class="highlight"><pre><span></span>$ python train.py
</pre></div>
</div>
<p>You should see the output as follows including the details of neural net structure with some parameter information, reading data files, and the performance details during training and testing process.</p>
<div class="highlight-default"><div class="highlight"><pre><span></span><span class="p">(</span><span class="mi">32</span><span class="n">L</span><span class="p">,</span> <span class="mi">32</span><span class="n">L</span><span class="p">,</span> <span class="mi">32</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">32</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">32</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">32</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">32</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">32</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">,</span> <span class="mi">16</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">32</span><span class="n">L</span><span class="p">,</span> <span class="mi">8</span><span class="n">L</span><span class="p">,</span> <span class="mi">8</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">32</span><span class="n">L</span><span class="p">,</span> <span class="mi">8</span><span class="n">L</span><span class="p">,</span> <span class="mi">8</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">64</span><span class="n">L</span><span class="p">,</span> <span class="mi">8</span><span class="n">L</span><span class="p">,</span> <span class="mi">8</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">64</span><span class="n">L</span><span class="p">,</span> <span class="mi">8</span><span class="n">L</span><span class="p">,</span> <span class="mi">8</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">64</span><span class="n">L</span><span class="p">,</span> <span class="mi">4</span><span class="n">L</span><span class="p">,</span> <span class="mi">4</span><span class="n">L</span><span class="p">)</span>
<span class="p">(</span><span class="mi">1024</span><span class="n">L</span><span class="p">,)</span>
<span class="n">Start</span> <span class="n">intialization</span><span class="o">............</span>
<span class="n">conv1_weight</span> <span class="n">gaussian</span> <span class="mf">7.938460476e-05</span>
<span class="n">conv1_bias</span> <span class="n">constant</span> <span class="mf">0.0</span>
<span class="n">conv2_weight</span> <span class="n">gaussian</span> <span class="mf">0.00793507322669</span>
<span class="n">conv2_bias</span> <span class="n">constant</span> <span class="mf">0.0</span>
<span class="n">conv3_weight</span> <span class="n">gaussian</span> <span class="mf">0.00799657031894</span>
<span class="n">conv3_bias</span> <span class="n">constant</span> <span class="mf">0.0</span>
<span class="n">dense_weight</span> <span class="n">gaussian</span> <span class="mf">0.00804364029318</span>
<span class="n">dense_bias</span> <span class="n">constant</span> <span class="mf">0.0</span>
<span class="n">Loading</span> <span class="n">data</span> <span class="o">..................</span>
<span class="n">Loading</span> <span class="n">data</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="n">py</span><span class="o">/</span><span class="n">data_batch_1</span>
<span class="n">Loading</span> <span class="n">data</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="n">py</span><span class="o">/</span><span class="n">data_batch_2</span>
<span class="n">Loading</span> <span class="n">data</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="n">py</span><span class="o">/</span><span class="n">data_batch_3</span>
<span class="n">Loading</span> <span class="n">data</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="n">py</span><span class="o">/</span><span class="n">data_batch_4</span>
<span class="n">Loading</span> <span class="n">data</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="n">py</span><span class="o">/</span><span class="n">data_batch_5</span>
<span class="n">Loading</span> <span class="n">data</span> <span class="n">file</span> <span class="n">cifar</span><span class="o">-</span><span class="mi">10</span><span class="o">-</span><span class="n">batches</span><span class="o">-</span><span class="n">py</span><span class="o">/</span><span class="n">test_batch</span>
<span class="n">Epoch</span> <span class="mi">0</span>
<span class="n">training</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.881866</span><span class="p">,</span> <span class="n">training</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.306360</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.420000</span>
<span class="n">test</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.602577</span><span class="p">,</span> <span class="n">test</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.412200</span>
<span class="n">Epoch</span> <span class="mi">1</span>
<span class="n">training</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.536011</span><span class="p">,</span> <span class="n">training</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.441940</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.500000</span>
<span class="n">test</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.378170</span><span class="p">,</span> <span class="n">test</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.507600</span>
<span class="n">Epoch</span> <span class="mi">2</span>
<span class="n">training</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.333137</span><span class="p">,</span> <span class="n">training</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.519960</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.520000</span>
<span class="n">test</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.272205</span><span class="p">,</span> <span class="n">test</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.540600</span>
<span class="n">Epoch</span> <span class="mi">3</span>
<span class="n">training</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.185212</span><span class="p">,</span> <span class="n">training</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.574120</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.540000</span>
<span class="n">test</span> <span class="n">loss</span> <span class="o">=</span> <span class="mf">1.211573</span><span class="p">,</span> <span class="n">test</span> <span class="n">accuracy</span> <span class="o">=</span> <span class="mf">0.567600</span>
</pre></div>
</div>
<p>This script will call <code class="docutils literal"><span class="pre">alexnet.py</span></code> file to build the alexnet model. After the training is finished, SINGA will save the model parameters into a checkpoint file <code class="docutils literal"><span class="pre">model.bin</span></code> in the same directory. Then we can use this <code class="docutils literal"><span class="pre">model.bin</span></code> file for prediction.</p>
<div class="highlight-default"><div class="highlight"><pre><span></span>$ python predict.py
</pre></div>
</div>
</div>
</div>
</div>
</div>
<footer>
<hr/>
<div role="contentinfo">
<p>
&copy; Copyright 2019 The Apache Software Foundation. All rights reserved. Apache SINGA, Apache, the Apache feather logo, and the Apache SINGA project logos are trademarks of The Apache Software Foundation. All other marks mentioned may be trademarks or registered trademarks of their respective owners.
</p>
</div>
Built with <a href="http://sphinx-doc.org/">Sphinx</a> using a <a href="https://github.com/rtfd/sphinx_rtd_theme">theme</a> provided by <a href="https://readthedocs.org">Read the Docs</a>.
</footer>
</div>
</div>
</section>
</div>
<script type="text/javascript">
jQuery(function () {
SphinxRtdTheme.Navigation.enable(true);
});
</script>
<div class="rst-versions" data-toggle="rst-versions" role="note" aria-label="versions">
<span class="rst-current-version" data-toggle="rst-current-version">
<span class="fa fa-book"> singa </span>
v: latest
<span class="fa fa-caret-down"></span>
</span>
<div class="rst-other-versions">
<dl>
<dt>Languages</dt>
<dd><a href=".././index.html">English</a></dd>
<dd><a href=".././zh/index.html">中文</a></dd>
</dl>
<dl>
<dt>Versions</dt>
<dd><a href="http://singa.apache.org/v0.3.0/">0.3</a></dd>
<dd><a href="http://singa.apache.org/v1.1.0/">1.1</a></dd>
</dl>
</div>
<a href="http://www.apache.org"
style="color:lightblue;padding: 5px; font-size: 10px; text-align: center; text-decoration: none; margin: 5px 2px;">Foundation</a>
<a href="http://www.apache.org/events/current-event"
style="color:lightblue;padding: 5px; font-size: 10px; text-align: center; text-decoration: none; margin: 5px 2px;">Events</a>
<a href="http://www.apache.org/foundation/thanks.html"
style="color:lightblue;padding: 5px; font-size: 10px; text-align: center; text-decoration: none; margin: 5px 2px;">Thanks</a>
<a href="http://www.apache.org/foundation/sponsorship.html"
style="color:lightblue;padding: 5px; font-size: 10px; text-align: center; text-decoration: none; margin: 5px 2px;">Sponsorship</a>
<a href="http://www.apache.org/licenses/"
style="color:lightblue;padding: 5px; font-size: 10px; text-align: center; text-decoration: none; margin: 5px 2px;">License</a>
<br>
<a href="https://github.com/apache/singa" class="fa fa-github"
style="padding: 10px; font-size: 20px; width: 30px; text-align: center; text-decoration: none; margin: 5px 2px;"></a>
<a href="https://aws.amazon.com/marketplace/seller-profile?id=5bcac385-12c4-4802-aec7-351e09b77b4c"
class="fab fa-aws"
style="padding: 10px; font-size: 20px; width: 30px; text-align: center; text-decoration: none; margin: 5px 2px;"></a>
<a href="https://hub.docker.com/r/apache/singa/" class="fab fa-docker"
style="padding: 10px; font-size: 20px; width: 30px; text-align: center; text-decoration: none; margin: 5px 2px;"></a>
<a href="https://www.linkedin.com/groups/13550034" class="fa fa-linkedin"
style="padding: 10px; font-size: 20px; width: 30px; text-align: center; text-decoration: none; margin: 5px 2px;"></a>
<a href="https://twitter.com/ApacheSinga" class="fa fa-twitter"
style="padding: 10px; font-size: 20px; width: 30px; text-align: center; text-decoration: none; margin: 5px 2px;"></a>
<a href="https://www.facebook.com/Apache-SINGA-347284219056544/" class="fa fa-facebook"
style="padding: 10px; font-size: 20px; width: 30px; text-align: center; text-decoration: none; margin: 5px 2px;"></a>
<a href="https://www.researchgate.net/project/Apache-SINGA" class="fab fa-researchgate"
style="padding: 10px; font-size: 20px; width: 30px; text-align: center; text-decoration: none; margin: 5px 2px;"></a>
</div>
<a href="https://github.com/apache/singa">
<img style="position: absolute; top: 0; right: 0; border: 0; z-index: 10000;"
src="https://s3.amazonaws.com/github/ribbons/forkme_right_orange_ff7600.png" alt="Fork me on GitHub">
</a>
</body>
</html>