Closed 41passerby closed 3 months ago
你的sam用的vary-tiny训练好的吗?
你的sam用的vary-tiny训练好的吗?
不是的,sam 没有加载训练好的权重,直接初始化的,我想自己训练一个 sam 。train_opt 的目标也是训练一个用于 ocr 的 sam 视觉模块对吧?因为也想在某些其他领域训练 sam 视觉模块,所以想学习下,感谢作者回复,麻烦再指导下,谢谢。
同样无法复现,我的loss倒是可以降低到1.8,但是还不够。测试中间checkpoint的输出只能学到图片的pattern,学不到具体的文字
@41passerby 请问你的问题解决了?
作者好,感谢分享 Vary 方案。
今天准备了一些 ocr 数据,依据说明训练 Vary-tiny ,发现 loss 一直降不下来,loss 维持 2.7 左右,请教下是什么问题导致,先行谢过。
基础模型使用的是 facebook/opt-125m。目前看数据样本挺正常的,如下截图,左边是文档图片,右边是 ocr 内容,每一行是一个段落。
数据格式:
loss 不降的截图: