You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

run_eval.sh 1.5 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. #!/bin/bash
  2. # Copyright 2020 Huawei Technologies Co., Ltd
  3. #
  4. # Licensed under the Apache License, Version 2.0 (the "License");
  5. # you may not use this file except in compliance with the License.
  6. # You may obtain a copy of the License at
  7. #
  8. # http://www.apache.org/licenses/LICENSE-2.0
  9. #
  10. # Unless required by applicable law or agreed to in writing, software
  11. # distributed under the License is distributed on an "AS IS" BASIS,
  12. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. # See the License for the specific language governing permissions and
  14. # limitations under the License.
  15. # ============================================================================
  16. if [ $# != 2 ]
  17. then
  18. echo "Usage: sh run_eval.sh [DATASET_PATH] [CHECKPOINT_PATH]"
  19. exit 1
  20. fi
  21. get_real_path(){
  22. if [ "${1:0:1}" == "/" ]; then
  23. echo "$1"
  24. else
  25. echo "$(realpath -m $PWD/$1)"
  26. fi
  27. }
  28. PATH1=$(get_real_path $1)
  29. PATH2=$(get_real_path $2)
  30. echo $PATH1
  31. echo $PATH2
  32. if [ ! -d $PATH1 ]
  33. then
  34. echo "error: DATASET_PATH=$PATH1 is not a directory"
  35. exit 1
  36. fi
  37. if [ ! -f $PATH2 ]
  38. then
  39. echo "error: CHECKPOINT_PATH=$PATH2 is not a file"
  40. exit 1
  41. fi
  42. ulimit -u unlimited
  43. export DEVICE_NUM=1
  44. export DEVICE_ID=0
  45. export RANK_SIZE=$DEVICE_NUM
  46. export RANK_ID=0
  47. if [ -d "eval" ];
  48. then
  49. rm -rf ./eval
  50. fi
  51. mkdir ./eval
  52. cp ../*.py ./eval
  53. cp *.sh ./eval
  54. cp -r ../src ./eval
  55. cd ./eval || exit
  56. env > env.log
  57. echo "start infering for device $DEVICE_ID"
  58. python eval.py --do_eval=True --dataset_path=$PATH1 --checkpoint_path=$PATH2 &> log &
  59. cd ..