added arg docs

This commit is contained in:
William Falcon
2019-07-18 12:05:52 -04:00
parent da842c0cd6
commit b1041220ac
@@ -116,6 +116,7 @@ class LightningTemplateModel(LightningModule):
:param outputs: list of individual outputs of each validation step
:return:
"""
print(outputs)
val_loss_mean = 0
val_acc_mean = 0
for output in outputs: