Spaces:
Build error
Build error
File size: 250 Bytes
471bf00 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
num_gpu=$1
cd $(dirname $(dirname $0))
exp_dir=$(pwd)
base_dir=$(dirname $(dirname $exp_dir))
config=${exp_dir}/config.json
export PYTHONPATH=$base_dir
export PYTHONIOENCODING=UTF-8
CUDA_VISIBLE_DEVICES=${num_gpu} python train.py -c config.json
|