add script for training

paddleseg/core/train.py

@@ -157,7 +157,7 @@ def train(model,
 
             if fp16:
                 with paddle.amp.auto_cast(
-                        enable=True, custom_black_list={'bilinear_interp_v2'}):
+                        enable=True, custom_white_list={"elementwise_add", "batch_norm"}, custom_black_list={'bilinear_interp_v2'}):
                     if nranks > 1:
                         logits_list = ddp_model(images)
                     else:

script/run_fp16.sh

@@ -0,0 +1,12 @@
+export FLAGS_conv_workspace_size_limit=2000 #MB
+export FLAGS_cudnn_exhaustive_search=1
+export FLAGS_cudnn_batchnorm_spatial_persistent=1
+
+python train.py --config benchmark/deeplabv3p.yml \
+  --iters=500 \
+  --batch_size 4 \
+  --learning_rate 0.01 \
+  --num_workers 8 \
+  --log_iters 20 \
+  --data_format NHWC \
+  --fp16

script/run_fp32.sh

@@ -0,0 +1,11 @@
+export FLAGS_conv_workspace_size_limit=2000 #MB
+export FLAGS_cudnn_exhaustive_search=1
+export FLAGS_cudnn_batchnorm_spatial_persistent=1
+
+python train.py --config benchmark/deeplabv3p.yml \
+  --iters=500 \
+  --batch_size 2 \
+  --learning_rate 0.01 \
+  --num_workers 8 \
+  --log_iters 20 \
+  --data_format NCHW \