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

added lightning model docs

Browse Source
This commit is contained in:
William Falcon
2019-06-27 10:04:24 -04:00
parent e2bcf1ecff
commit c0c2e644fd
3 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/example_model.py
-4
View File
@@ -71,9 +71,6 @@ class ExampleModel(RootModule):
# calculate loss
loss_val = self.loss(y, y_hat)
# tqdm_dic = {'tng_loss': loss_val.item()}
# return loss_val, tqdm_dic
output = OrderedDict({
'loss': loss_val,
'tqdm_metrics': {}
@@ -96,7 +93,6 @@ class ExampleModel(RootModule):
labels_hat = torch.argmax(y_hat, dim=1)
val_acc = torch.sum(y == labels_hat).item() / (len(y) * 1.0)
# output = {'y_hat': y_hat, 'val_loss': loss_val.item(), 'val_acc': val_acc}
output = OrderedDict({
'val_loss': loss_val,
'val_acc': torch.tensor(val_acc),