wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Deployed 58d52c2 with MkDocs version: 1.0.4

Browse Source
This commit is contained in:
William Falcon
2019-10-18 23:57:14 +01:00
parent 0675db8074
commit 65095b6cac
43 changed files with 518 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files
LightningModule/RequiredTrainerInterface/index.html
+18 -8
View File
@@ -34,7 +34,7 @@
<meta name="lang:search.tokenizer" content="[\s\-]+">
<link rel="shortcut icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.0.4, mkdocs-material-4.4.2">
<meta name="generator" content="mkdocs-1.0.4, mkdocs-material-4.4.3">
@@ -1201,17 +1201,27 @@
12
13
14
15</pre></div></td><td class="code"><div class="codehilite"><pre><span></span><span class="k">def</span> <span class="nf">training_step</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">batch</span><span class="p">,</span> <span class="n">batch_nb</span><span class="p">):</span>
15
16
17
18
19
20</pre></div></td><td class="code"><div class="codehilite"><pre><span></span><span class="k">def</span> <span class="nf">training_step</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">batch</span><span class="p">,</span> <span class="n">batch_nb</span><span class="p">):</span>
<span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">,</span> <span class="n">z</span> <span class="o">=</span> <span class="n">batch</span>
<span class="c1"># implement your own</span>
<span class="n">out</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">forward</span><span class="p">(</span><span class="n">x</span><span class="p">)</span>
<span class="n">loss</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">loss</span><span class="p">(</span><span class="n">out</span><span class="p">,</span> <span class="n">x</span><span class="p">)</span>
<span class="n">logger_logs</span> <span class="o">=</span> <span class="p">{</span><span class="s1">&#39;training_loss&#39;</span><span class="p">:</span> <span class="n">loss</span><span class="p">}</span> <span class="c1"># optional (MUST ALL BE TENSORS)</span>
<span class="c1"># if using TestTubeLogger or TensorboardLogger you can nest scalars</span>
<span class="n">logger_logs</span> <span class="o">=</span> <span class="p">{</span><span class="s1">&#39;losses&#39;</span><span class="p">:</span> <span class="n">logger_logs</span><span class="p">}</span> <span class="c1"># optional (MUST ALL BE TENSORS)</span>
<span class="n">output</span> <span class="o">=</span> <span class="p">{</span>
<span class="s1">&#39;loss&#39;</span><span class="p">:</span> <span class="n">loss</span><span class="p">,</span> <span class="c1"># required</span>
<span class="s1">&#39;progress_bar&#39;</span><span class="p">:</span> <span class="p">{</span><span class="s1">&#39;training_loss&#39;</span><span class="p">:</span> <span class="n">loss</span><span class="p">},</span> <span class="c1"># optional (MUST ALL BE TENSORS)</span>
<span class="s1">&#39;log&#39;</span><span class="p">:</span> <span class="p">{</span><span class="s1">&#39;training_loss&#39;</span><span class="p">:</span> <span class="n">loss</span><span class="p">}</span> <span class="c1"># optional (MUST ALL BE TENSORS)</span>
<span class="s1">&#39;log&#39;</span><span class="p">:</span> <span class="n">logger_logs</span>
<span class="p">}</span>
<span class="c1"># return a dict</span>
@@ -1456,10 +1466,10 @@ If you don't need to validate you don't need to implement this method. In this s
</pre></div>
</td></tr></table>
<p>If you didn't define a validation_step, this won't be called. </p>
<p>Called at the end of the validation loop with the outputs of validation_step.</p>
<p>The outputs here are strictly for the progress bar. If you don't need to display anything, don't return anything. </p>
<p><strong>Params</strong> </p>
<p>If you didn't define a validation_step, this won't be called. Called at the end of the validation loop with the outputs of validation_step.</p>
<p>The outputs here are strictly for the progress bar. If you don't need to display anything, don't return anything. <br />
Any keys present in 'log', 'progress_bar' or the rest of the dictionary are available for callbacks to access.
<strong>Params</strong> </p>
<table>
<thead>
<tr>
@@ -2114,7 +2124,7 @@ The <a href="https://williamfalcon.github.io/test-tube/hyperparameter_optimizati
</div>
<script src="../../assets/javascripts/application.c648116f.js"></script>
<script src="../../assets/javascripts/application.ac79c3b0.js"></script>
<script>app.initialize({version:"1.0.4",url:{base:"../.."}})</script>