!4055 add docker_start.sh

Merge pull request !4055 from FangManLin/master

research/cv/PSPNet/docker_start.sh

+#!/bin/bash
+
+docker_image=$1
+data_dir=$2
+model_dir=$3
+
+docker run -it --ipc=host \
+               --device=/dev/davinci0 \
+               --device=/dev/davinci1 \
+               --device=/dev/davinci2 \
+               --device=/dev/davinci3 \
+               --device=/dev/davinci4 \
+               --device=/dev/davinci5 \
+               --device=/dev/davinci6 \
+               --device=/dev/davinci7 \
+               --device=/dev/davinci_manager \
+               --device=/dev/devmm_svm --device=/dev/hisi_hdc \
+               -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
+               -v /usr/local/Ascend/add-ons/:/usr/local/Ascend/add-ons/ \
+               -v ${model_dir}:${model_dir} \
+               -v ${data_dir}:${data_dir}  \
+               -v /var/log/npu/conf/slog/:/var/log/npu/conf/slog/ \
+               -v /var/log/npu/slog/:/var/log/npu/slog -v /var/log/npu/profiling/:/var/log/npu/profiling \
+               -v /var/log/npu/dump/:/var/log/npu/dump -v /var/log/npu/:/usr/slog ${docker_image} \
+               /bin/bash
\ No newline at end of file

research/cv/PSPNet/modelarts/train_start.py

@@ -16,6 +16,8 @@
 import os
 import ast
 import argparse
+import subprocess
+
 from src.utils import functions_args as fa
 from src.model import pspnet
 from src.model.cell import Aux_CELoss_Cell