Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feature/5275 clean progress bar print #5470

Merged
merged 17 commits into from
Feb 22, 2021
Merged

Feature/5275 clean progress bar print #5470

Changes from 1 commit
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
85ea319
Trainer.test should return only test metrics (#5214)
tchaton Dec 28, 2020
5c112f0
Fix metric state reset (#5273)
tadejsv Dec 29, 2020
2072b61
print() method added to ProgressBar
Jan 10, 2021
539e2e6
printing alongside progress bar added to LightningModule.print()
Jan 10, 2021
023993b
LightningModule.print() method documentation updated
Jan 10, 2021
dbba2d7
ProgressBarBase.print() stub added
Jan 15, 2021
1bfc866
stub
rohitgr7 Jan 19, 2021
0ee4abc
add progress bar tests
awaelchli Jan 20, 2021
a70b7c0
fix isort
awaelchli Jan 20, 2021
a13a218
Progress Callback fixes
Feb 3, 2021
19411f7
test_metric.py duplicate DummyList removed
Feb 8, 2021
68be2b8
PEP and isort fixes
Feb 8, 2021
ab908cf
CHANGELOG updated
Feb 8, 2021
599aa6a
test_progress_bar_print win linesep fix
Feb 8, 2021
b9d61c6
Merge branch 'master' into feature/5275_clean_progress_bar_print
asnorkin Feb 18, 2021
331aacc
test_progress_bar.py remove whitespaces
Feb 18, 2021
c152e0c
Update CHANGELOG.md
carmocca Feb 18, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
printing alongside progress bar added to LightningModule.print()
  • Loading branch information
Alexander Snorkin authored and Alexander Snorkin committed Feb 17, 2021
commit 539e2e660db09987fcac8046f18389055ab3d671
6 changes: 5 additions & 1 deletion pytorch_lightning/core/lightning.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,11 @@ def forward(self, x):

"""
if self.trainer.is_global_zero:
print(*args, **kwargs)
progress_bar = self.trainer.progress_bar_callback
if progress_bar is not None and progress_bar.is_enabled:
progress_bar.print(*args, **kwargs)
else:
print(*args, **kwargs)

def log(
self,
Expand Down