From cbb2427f0d263819b1a637477410d3b85ab6bc2d Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 25 Jun 2020 18:54:32 -0400 Subject: [PATCH] changed apex level (#2362) --- pytorch_lightning/trainer/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 712a11fd37..9d1736e6eb 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -126,7 +126,7 @@ class Trainer( terminate_on_nan: bool = False, auto_scale_batch_size: Union[str, bool] = False, 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 use_amp=None, # backward compatible, todo: remove in v0.9.0 show_progress_bar=None, # backward compatible, todo: remove in v0.9.0