From 64b606a3cd2bde6cdb2782438f1fd9fab9bf3d2b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 16 Jul 2019 18:49:54 +0200 Subject: [PATCH] updates --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index 98424ed6..b1d204f8 100644 --- a/train.py +++ b/train.py @@ -156,6 +156,7 @@ def train(cfg, try: from apex import amp model, optimizer = amp.initialize(model, optimizer, opt_level='O1', verbosity=0) + print('Using Apex') except: # not installed: install help: https://github.com/NVIDIA/apex/issues/259 mixed_precision = False