From 4849324629994aa719d09d27f6b851986fda7044 Mon Sep 17 00:00:00 2001
From: mzr1996 <mzr1996@163.com>
Date: Wed, 11 Oct 2023 11:12:32 +0800
Subject: [PATCH] Update train.py to compat with new config

---
 tools/train.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/tools/train.py b/tools/train.py
index 84c1eec93aa..89c8548fc32 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -91,10 +91,6 @@ def merge_args(cfg, args):
 
     # enable automatic-mixed-precision training
     if args.amp is True:
-        optim_wrapper = cfg.optim_wrapper.get('type', 'OptimWrapper')
-        assert optim_wrapper in ['OptimWrapper', 'AmpOptimWrapper'], \
-            '`--amp` is not supported custom optimizer wrapper type ' \
-            f'`{optim_wrapper}.'
         cfg.optim_wrapper.type = 'AmpOptimWrapper'
         cfg.optim_wrapper.setdefault('loss_scale', 'dynamic')