From fecd6a00cb93df99c0c8612761b4105cac07feda Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 13 May 2019 20:41:23 -0400 Subject: [PATCH] added option to change default tensor --- pytorch_lightning/models/trainer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index ee47791ef6..9cafea886f 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -81,6 +81,8 @@ class Trainer(TrainerIO): # apex test use_amp = True self.use_amp = use_amp and APEX_AVAILABLE + if self.use_amp: + print('using 16bit precision') def __determine_data_use_amount(self, train_percent_check, val_percent_check, test_percent_check, overfit_pct): """