update readme

gh-pages
KaiyangZhou 2019-03-24 21:01:42 +00:00
parent 808f7563e1
commit 3467757cef
2 changed files with 2 additions and 2 deletions

View File

@ -240,7 +240,7 @@ $ <span class="c1"># python setup.py develop</span>
<span class="n">pretrained</span><span class="o">=</span><span class="bp">True</span>
<span class="p">)</span>
<span class="n">model</span> <span class="o">=</span> <span class="n">model</span><span class="o">.</span><span class="n">cuda</span><span class="p">()</span> <span class="c1"># move model to gpu</span>
<span class="n">model</span> <span class="o">=</span> <span class="n">model</span><span class="o">.</span><span class="n">cuda</span><span class="p">()</span>
<span class="n">optimizer</span> <span class="o">=</span> <span class="n">torchreid</span><span class="o">.</span><span class="n">optim</span><span class="o">.</span><span class="n">build_optimizer</span><span class="p">(</span>
<span class="n">model</span><span class="p">,</span>

File diff suppressed because one or more lines are too long