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

Set warnings : Unify epoch numbers to be zero-based : #675 (#786)

Browse Source 
* [update] : #675 : set warnings

* [fix] : #675 : remove white space
This commit is contained in:
Shunsuke Hidaka
2020-02-05 05:15:51 -05:00
committed by GitHub
parent 4cbcb7887e
commit 734b28ed2d
2 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/training_loop.py
+2
View File
@@ -281,6 +281,8 @@ class TrainerTrainLoopMixin(ABC):
pass
def train(self):
warnings.warn('Displayed epoch numbers in the progress bar start from "1" until v0.6.x,'
' but will start from "0" in v0.8.0.', DeprecationWarning)
model = self.get_model()
# run all epochs
for epoch in range(self.current_epoch, self.max_epochs):