finetune_funsd_sentence.py 420 Bytes
Newer Older
HHL's avatar
v  
HHL committed
1
2
3
4
5
6
7
8
9
from libs.utils.vocab import FunsdEntityVocab

train_lrc_paths = ["/yrfs1/intern/pfhu6/GraphDoc/dataset/funsd/train.lrc"]
train_image_dirs = ["/yrfs1/intern/pfhu6/GraphDoc/dataset/funsd/dataset/training_data/images/"]

valid_lrc_paths = ["/yrfs1/intern/pfhu6/GraphDoc/dataset/funsd/val.lrc"]
valid_image_dirs = ["/yrfs1/intern/pfhu6/GraphDoc/dataset/funsd/dataset/testing_data/images/"]

funsd_vocab = FunsdEntityVocab()