This commit is contained in:
Starbeamrainbowlabs 2022-10-12 17:33:54 +01:00
parent dbe4fb0eab
commit 7933564c66
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -63,7 +63,7 @@ echo ">>> Installing requirements";
conda run -n py38 pip install -r requirements.txt;
echo ">>> Training model";
#shellcheck disable=SC2086
/usr/bin/env time -v conda run -n py38 src/index.py pretrain --input "${INPUT}" --output "${dir_output}" ${ARGS};
/usr/bin/env time -v conda run -n py38 src/index.py train --input "${INPUT}" --output "${dir_output}" ${ARGS};
src/index.py train --input "${INPUT}" -o "${dir_output}"
# Recommended extra args: --water-size 94 --batch-size 48 --arch convnext_i_xtiny