Update train.py to compat with new config

open-mmlab · Oct 11, 2023 · 4849324 · 4849324
1 parent b0a792e
commit 4849324
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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')