适配训练脚本

配置训练脚本，以将训练日志重定向至指定目录。以训练脚本train.py为例，执行参考样例如下：

python train.py &> "${TRAIN_LOG_DIR}/rank-{npu id}.log"
wait