From 7d237d820c25e857a980bde29e20f05b7e5cd02c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=89=E6=B4=8B=E4=B8=89=E6=B4=8B?= <1258009915@qq.com> Date: Fri, 12 Apr 2024 03:20:04 +0000 Subject: [PATCH] work in progress --- src/models/ocr_model/train/train.py | 2 +- src/start_web.sh | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/models/ocr_model/train/train.py b/src/models/ocr_model/train/train.py index 25671de..9a0bc8a 100644 --- a/src/models/ocr_model/train/train.py +++ b/src/models/ocr_model/train/train.py @@ -96,7 +96,7 @@ if __name__ == '__main__': collate_fn_with_tokenizer = partial(collate_fn, tokenizer=tokenizer) # model = TexTeller() - model = TexTeller.from_pretrained('/home/lhy/code/TexTeller/src/models/ocr_model/model/ckpt') + model = TexTeller.from_pretrained('/home/lhy/code/TexTeller/src/models/ocr_model/train/train_result/TexTellerv3/checkpoint-440000') # ================= debug ======================= # foo = train_dataset[:50] diff --git a/src/start_web.sh b/src/start_web.sh index 3d700cc..7dab5d2 100755 --- a/src/start_web.sh +++ b/src/start_web.sh @@ -1,8 +1,8 @@ #!/usr/bin/env bash set -exu -# export CHECKPOINT_DIR="/home/lhy/code/TexTeller/src/models/ocr_model/model/ckpt" -export CHECKPOINT_DIR="default" +export CHECKPOINT_DIR="/home/lhy/code/TexTeller/src/models/ocr_model/train/train_result/TexTellerv3/checkpoint-460000" +# export CHECKPOINT_DIR="default" export TOKENIZER_DIR="/home/lhy/code/TexTeller/src/models/tokenizer/roberta-tokenizer-7Mformulas" export USE_CUDA=True # True or False (case-sensitive) export NUM_BEAM=3