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

adding framework level dp

Browse Source
This commit is contained in:
William Falcon
2019-06-25 18:10:15 -04:00
parent 516f441153
commit 6c705a0525
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/example_model.py
+2 -2
View File
@@ -55,7 +55,7 @@ class ExampleModel(RootModule):
nll = F.nll_loss(logits, labels)
return nll
def training_step(self, data_batch):
def training_step(self, data_batch, batch_i):
"""
Called inside the training loop
:param data_batch:
@@ -72,7 +72,7 @@ class ExampleModel(RootModule):
tqdm_dic = {'tng_loss': loss_val.item()}
return loss_val, tqdm_dic
def validation_step(self, data_batch):
def validation_step(self, data_batch, batch_i):
"""
Called inside the validation loop
:param data_batch: