Merge pull request #912 from wuyefeilin/benchmark_amp_dis

PaddlePaddle · Mar 23, 2021 · 15d04a6 · 15d04a6
2 parents e90a53e + cc8b349
commit 15d04a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -132,7 +132,7 @@ def main(args):
 
     # Only support for the DeepLabv3+ model
     if args.data_format == 'NHWC':
-        if cfg.dic['model'] != 'DeepLabV3P':
+        if cfg.dic['model']['type'] != 'DeepLabV3P':
             raise ValueError(
                 'The "NHWC" data format only support the DeepLabV3P model!')
         cfg.dic['model']['data_format'] = args.data_format