Commit Graph

15 Commits

Author SHA1 Message Date
三洋三洋
6373e19132 merge v2 2024-03-26 08:16:28 +00:00
三洋三洋
b293026f5e 数据增强加入了是否只针对扫描件的选项 2024-03-26 07:37:48 +00:00
三洋三洋
ef7cccff03 TexTellerv2 2024-03-25 11:46:43 +00:00
三洋三洋
93979bddf6 修复了bug:当样本中出现非常长的公式(对应的token数可能超过2048),会导致给label进行embedding时index out of range 2024-03-06 13:59:36 +00:00
三洋三洋
04a4bddd97 修改了functional.py中的tokenize_fn(图片以png形式而非数组的形式保存,防止占用过多硬盘空间) 2024-03-04 05:38:30 +00:00
三洋三洋
2d6c46b88d 修改好了训练,加入了数据增强 2024-03-04 05:35:59 +00:00
三洋三洋
04b99b8451 完成了web,ray server,重构了代码 2024-02-08 13:55:22 +00:00
三洋三洋
b5dbf64716 把代码修改成了接受输入为png的图片 2024-02-03 09:40:13 +00:00
三洋三洋
ab1a05bf32 完成了所有代码 2024-02-02 04:50:19 +00:00
三洋三洋
ebac28a90d tmp commit 2024-01-31 10:44:41 +00:00
三洋三洋
1fba652766 Initial Commit 2024-01-31 10:11:07 +00:00
三洋三洋
b7bf5c444f 加入和推理和评估的代码 2024-01-30 08:36:23 +00:00
三洋三洋
14125da26f 1) 加入了推理代码; 2) 整理了其他代码 2024-01-28 14:03:42 +00:00
三洋三洋
c6d5c91955 写完了模型代码、Tokenizer、数据预处理、训练脚本,但目前的训练脚本没有配置generate(评估仅能看loss) 2024-01-28 06:19:23 +00:00
三洋三洋
9d27ee0585 写好了ocr_model训练脚本的大致框架 2024-01-23 04:23:08 +00:00