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