From 9bb7a30f39fb7786e1576ab667c1ac3785abc46a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 20:03:31 -0400 Subject: [PATCH] scaled batch size --- pytorch_lightning/models/trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index a85fbafe..9da7024a 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -142,9 +142,10 @@ class Trainer(TrainerIO): You set use_amp=True but do not have apex installed. Install apex first using this guide and rerun with use_amp=True: https://github.com/NVIDIA/apex#linux + + this run will NOT use 16 bit precision ''' warnings.warn(msg) - print('not using 16 bit') def __determine_data_use_amount(self, train_percent_check, val_percent_check, test_percent_check, overfit_pct): """