Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

set the default value of progress_bar_refresh_rate to 1 (#1100)

This commit is contained in:
So Uchida 2020-03-13 01:40:30 +09:00 committed by GitHub
parent ed0b890e75
commit 322e7157e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pytorch_lightning/trainer/__init__.py
View File

@ -619,13 +619,13 @@ Example::
progress_bar_refresh_rate progress_bar_refresh_rate
^^^^^^^^^^^^^^^^^^^^^^^^^ ^^^^^^^^^^^^^^^^^^^^^^^^^
How often to refresh progress bar (in steps). How often to refresh progress bar (in steps).
Faster refresh rates (lower number), in notebooks is known to crash them In notebooks, faster refresh rates (lower number) is known to crash them
because of their screen refresh rates. 50 is optimal for those cases. because of their screen refresh rates, so raise it to 50 or more.
Example:: Example::
# default used by the Trainer # default used by the Trainer
trainer = Trainer(progress_bar_refresh_rate=50) trainer = Trainer(progress_bar_refresh_rate=1)
reload_dataloaders_every_epoch reload_dataloaders_every_epoch

2
pytorch_lightning/trainer/trainer.py
View File

@ -88,7 +88,7 @@ class Trainer(
num_tpu_cores: Optional[int] = None, num_tpu_cores: Optional[int] = None,
log_gpu_memory: Optional[str] = None, log_gpu_memory: Optional[str] = None,
show_progress_bar: bool = True, show_progress_bar: bool = True,
progress_bar_refresh_rate: int = 50, progress_bar_refresh_rate: int = 1,
overfit_pct: float = 0.0, overfit_pct: float = 0.0,
track_grad_norm: int = -1, track_grad_norm: int = -1,
check_val_every_n_epoch: int = 1, check_val_every_n_epoch: int = 1,