Merge pull request #43 from Frederisk/patch-1

Convert `MODEL_DIR` to an absolute path
This commit is contained in:
Brendan Dolan-Gavitt 2022-09-05 18:14:53 -04:00 committed by GitHub
commit 7de9d215f9
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