Skip to content

Commit 2e89e4b

Browse files
committed
checkpoints format is now tflite
1 parent ca0d7f6 commit 2e89e4b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

STT/train.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@ pushd $STT_DIR
99

1010
mkdir -p /mnt/sources/feature_cache || true
1111

12-
# Do not overwrite checkpoint file if model already exist: we will likely
12+
# Do not overwrite checkpoint file if model already exist: we will likely
1313
# only package
14-
if [ -f "/transfer-checkpoint/checkpoint" -a ! -f "/mnt/models/output_graph.pb" ]; then
14+
if [ -f "/transfer-checkpoint/checkpoint" -a ! -f "/mnt/models/output_graph.tflite" ]; then
1515
echo "Using checkpoint from ${TRANSFER_CHECKPOINT}"
1616
cp -a /transfer-checkpoint/* /mnt/checkpoints/
1717
fi;

0 commit comments

Comments
 (0)