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

changed apex level (#2362)

This commit is contained in:
William Falcon 2020-06-25 18:54:32 -04:00 committed by GitHub
parent 0a092f6683
commit cbb2427f0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/trainer/trainer.py
View File

@ -126,7 +126,7 @@ class Trainer(
terminate_on_nan: bool = False, terminate_on_nan: bool = False,
auto_scale_batch_size: Union[str, bool] = False, auto_scale_batch_size: Union[str, bool] = False,
prepare_data_per_node: bool = True, prepare_data_per_node: bool = True,
amp_level: str = 'O1', # backward compatible, todo: remove in v1.0.0 amp_level: str = 'O2', # backward compatible, todo: remove in v1.0.0
num_tpu_cores: Optional[int] = None, # backward compatible, todo: remove in v0.9.0 num_tpu_cores: Optional[int] = None, # backward compatible, todo: remove in v0.9.0
use_amp=None, # backward compatible, todo: remove in v0.9.0 use_amp=None, # backward compatible, todo: remove in v0.9.0
show_progress_bar=None, # backward compatible, todo: remove in v0.9.0 show_progress_bar=None, # backward compatible, todo: remove in v0.9.0