From 05ec8b3b9d505829429ed56658be172517c6b5be Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 18 Apr 2019 16:45:38 +0200 Subject: [PATCH] updates --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index 3ca70760..4d2b5c15 100644 --- 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