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

Deployed 087be2f with MkDocs version: 1.0.4

Browse Source
This commit is contained in:
William Falcon
2019-08-13 12:02:33 -05:00
parent 25bf79b9da
commit 5856805d27
3 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
LightningModule/RequiredTrainerInterface/index.html
+9
View File
@@ -1118,6 +1118,15 @@ class CoolModel(pl.LightningModule):
return output
</code></pre>
<p>If you define multiple optimizers, this step will also be called with an additional <code>optimizer_idx</code> param. </p>
<pre><code class="python"># Multiple optimizers (ie: GANs)
def training_step(self, data_batch, batch_nb, optimizer_idx):
if optimizer_idx == 0:
# do training_step with encoder
if optimizer_idx == 1:
# do training_step with decoder
</code></pre>
<hr />
<h3 id="tng_dataloader">tng_dataloader</h3>
<pre><code class="python">@pl.data_loader