search

HarderThenHarder / transformers_tasks

⭐️ NLP Algorithms with transformers lib. Supporting Text-Classification, Text-Generation, Information-Extraction, Text-Matching, RLHF, SFT etc.
https://www.zhihu.com/column/c_1451236880973426688
2.11k stars 376 forks source link

UI界面打开提示模型未被定义 #43

Open kbwzy opened 1 year ago

kbwzy commented 1 year ago

微调完模型，设置好了model_path,启动界面后，界面启动成功，但是点generate会报错： image

HarderThenHarder commented 1 year ago

Hi，您尝试下将 这行 代码进行以下修改：

out = model.generate(...) -> out = st.session_state['model'].generate(...)
kbwzy commented 1 year ago

已解决，谢谢！

kbwzy commented 1 year ago

还有一个问题，我这边训练效果没有你演示界面效果好，是不是因为训练数据不足导致的啊？目前只有912条训练数据

HarderThenHarder commented 1 year ago

这有可能受到调参的影响，我没有使用额外的数据进行训练。 如果需要，您也可以在 这里 找到更多数据进行微调。

kbwzy commented 1 year ago

@HarderThenHarder 你的意思是更有可能是调参的问题。我这边是P-Turning的默认参数，去掉了量化。这边效果是这样的： image 第一：好像还不理解SPO三元组，返回并不是一个个的三元组tuple 第二：没有返回训练集中的\"predicate\": \"作者\", \"object_type\": \"人物\", \"subject_type\": \"图书作品\", \"object\": \"季静好\", \"subject\": \"安得君顾\"这些属性key。 我接下来怎么调参可以有改进呢？

fanyongchao01 commented 1 year ago

@kbwzy 你训练了多久， image

我都跑了一个晚上了 还没有跑完