# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
# 
# 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.

export BASE_DATA_DIR="/home/users/mlperf-workspace/bert_data" 
export NEXP="${2:-"400"}"

export STAGE=${STAGE:-"run"}
export CONT="nvcr.io/nvidia/pytorch:22.04-py3-paddle-dev-test"
export CLEAR_CACHES=1

if [[ $STAGE == "build" ]]; then
  bash -ex Dockerfiles/build_with_pip_install_whl.sh  
else
  # rm -rf results
  bash -ex run_with_docker.sh
fi