updates

2019-07-16 18:49:54 +02:00 · 2019-07-16 18:49:54 +02:00 · 64b606a3cd
commit 64b606a3cd
parent 51d7e460a3
1 changed files with 1 additions and 0 deletions
--- 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