updates

2019-04-14 16:00:04 +02:00 · 2019-04-14 16:00:04 +02:00 · 09949cdafa
commit 09949cdafa
parent 52464f5a06
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -101,7 +101,7 @@ def train(
    mixed_precision = False
    if mixed_precision:
        from apex import amp
-        model, optimizer = amp.initialize(model, optimizer, opt_level='01')
+        model, optimizer = amp.initialize(model, optimizer, opt_level='O1')

    # Start training
    t = time.time()