#!/bin/bash# Copyright 2020 Huawei Technologies Co., Ltd## Licensed under the Apache License, Version 2.0 (the "License");# you may not use this file except in compliance with the License.# You may obtain a copy of the License at## http://www.apache.org/licenses/LICENSE-2.0## Unless required by applicable law or agreed to in writing, software# distributed under the License is distributed on an "AS IS" BASIS,# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.# See the License for the specific language governing permissions and# limitations under the License.# ============================================================================echo"=============================================================================================================="echo"Please run the script as: "echo"bash run_distribute_train_gpu.sh DATA_PATH"echo"for example: bash run_distribute_train_gpu.sh /path/ImageNet2012/train"echo"=============================================================================================================="DATA_PATH=$1mpirun -n 8 python train.py \--device_target="GPU"\--dataset="imagenet2012"\--is_distributed=1 \--data_path=$DATA_PATH> output.train.log 2>&1 &
