From ff2a21a08a153d0da133887a158a37eb1b307301 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 8 Oct 2019 09:09:57 -0400 Subject: [PATCH] default to O1 (#334) --- 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 d36ed26f..c9fb5077 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -86,7 +86,7 @@ class Trainer(TrainerIO): print_nan_grads=False, print_weights_summary=True, weights_save_path=None, - amp_level='O2', + amp_level='O1', nb_sanity_val_steps=5): """