Convert MODEL_DIR to an absolute path

To fix issue #41.
This commit is contained in:
Rowe Wilson Frederisk Holme 2022-09-06 02:51:06 +08:00 committed by GitHub
commit 704ef5ad58
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -39,6 +39,8 @@ NUM_GPUS=${NUM_GPUS:-1}
read -p "Where do you want to save the model [$(pwd)/models]? " MODEL_DIR
if [ -z "$MODEL_DIR" ]; then
MODEL_DIR="$(pwd)/models"
else
MODEL_DIR="$(readlink -m "${MODEL_DIR}")"
fi
# Write config.env