updates

2019-04-18 16:45:38 +02:00 · 2019-04-18 16:45:38 +02:00 · 05ec8b3b9d
parent c68093ee7d
commit 05ec8b3b9d
1 changed files with 1 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -120,6 +120,7 @@ def train(
                            sampler=sampler)

    # Mixed precision training https://github.com/NVIDIA/apex
+    # install help: https://github.com/NVIDIA/apex/issues/259
    mixed_precision = False
    if mixed_precision:
        from apex import amp