qurator-spk/train-calamari-gt4histocr
1
0
Fork 0
mirror of https://github.com/qurator-spk/train-calamari-gt4histocr.git synced 2025-06-08 19:30:02 +02:00
Code Issues Releases Wiki Activity

💩 calamari-models/train-calamari-gt4histocr: Force TF memory growth

Browse source
This commit is contained in:
Gerber, Mike 2019-12-10 17:57:00 +01:00
parent d5ed15a35a
commit aadb3f5a53
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
train.sh
View file

@ -51,6 +51,8 @@ export PYTHONUNBUFFERED=1 # For python + tee
outdir=$DATA_SUBDIR/calamari-models/GT4HistOCR outdir=$DATA_SUBDIR/calamari-models/GT4HistOCR
mkdir -p $outdir mkdir -p $outdir
export TF_FORCE_GPU_ALLOW_GROWTH=true # To prevent TF from taking all GPU memory
calamari-cross-fold-train \ calamari-cross-fold-train \
--files \ --files \
"$TMPDIR/*/*/*.png" \ "$TMPDIR/*/*/*.png" \
@ -59,6 +61,6 @@ calamari-cross-fold-train \
--early_stopping_nbest=5 \ --early_stopping_nbest=5 \
--batch_size=128 \ --batch_size=128 \
--n_folds=5 \ --n_folds=5 \
--max_parallel_models=1 \ --max_parallel_models=3 \
--display=0.01 \ --display=0.01 \
2>&1 | tee $outdir/train.`date -Iminutes`.log 2>&1 | tee $outdir/train.`date -Iminutes`.log