From d2284b373f26a1159c79c24b4dcb3017e336813f Mon Sep 17 00:00:00 2001
From: xieenze <Jonnhy_ez@163.com>
Date: Wed, 9 Sep 2020 16:56:55 +0800
Subject: [PATCH] update

---
 openselfsup/apis/train.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/openselfsup/apis/train.py b/openselfsup/apis/train.py
index a1da0a90..4314f318 100644
--- a/openselfsup/apis/train.py
+++ b/openselfsup/apis/train.py
@@ -248,9 +248,12 @@ def _non_dist_train(model,
             seed=cfg.seed,
             drop_last=getattr(cfg.data, 'drop_last', False)) for ds in dataset
     ]
+
+    if 'use_fp16' in cfg and cfg.use_fp16 == True:
+        raise NotImplementedError('apex do not support non_dist_trian!')
     # put model on gpus
     model = MMDataParallel(model, device_ids=range(cfg.gpus)).cuda()
-
+    
     # build runner
     optimizer = build_optimizer(model, cfg.optimizer)
     runner = Runner(