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

decouple returns from each step (#307)

Browse Source 
* decoupled training metrics from logging metrics

* decoupled validation metrics from log metrics

* updated docs

* updated docs

* updated docs

* Fixed test

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master

* merged master
This commit is contained in:
William Falcon
2019-10-05 13:35:20 -04:00
committed by GitHub
parent 8f5a06bfb8
commit 6cc3f1757f
9 changed files with 193 additions and 248 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/hooks.py
-3
View File
@@ -28,9 +28,6 @@ class ModelHooks(torch.nn.Module):
def on_post_performance_check(self):
pass
def on_training_metrics(self, metrics):
pass
def on_before_zero_grad(self, optimizer):
"""
Called after optimizer.step() and before optimizer.zero_grad()