Code Monkey home page Code Monkey logo

Comments (8)

hiyouga avatar hiyouga commented on May 18, 2024 2

抱歉,我的说法可能有误,我重新参考了 Vicuna 的训练代码,这种方式的确能加速模型在多轮对话上的训练,我们考虑在近期实现类似的功能,感谢你的建议!

from llama-factory.

hiyouga avatar hiyouga commented on May 18, 2024

目前的模型训练支持多轮对话,需要在 dataset_info.json 中指定 history 列。
在多轮对话的训练中,目前普遍采用的方式是

q1 + a1 + q2 + a2 + q3 + a3
[IGNORE] + [IGNORE] + [IGNORE] + [IGNORE] + [IGNORE] + a3

因此目前的实现方式适配多轮对话训练。

from llama-factory.

nieallen avatar nieallen commented on May 18, 2024

目前的模型训练支持多轮对话,需要在 dataset_info.json 中指定 history 列。 在多轮对话的训练中,目前普遍采用的方式是

q1 + a1 + q2 + a2 + q3 + a3
[IGNORE] + [IGNORE] + [IGNORE] + [IGNORE] + [IGNORE] + a3

因此目前的实现方式适配多轮对话训练。

多轮语料,每一轮只遮挡q,不遮挡a,会不会更好,让模型学到每一轮的回答,帮助更好做对话

from llama-factory.

hiyouga avatar hiyouga commented on May 18, 2024

这可能会破坏掉 BOS 和 EOS 的语义信息,我们不推荐这么做。

from llama-factory.

nieallen avatar nieallen commented on May 18, 2024

抱歉,我的说法可能有误,我重新参考了 Vicuna 的训练代码,这种方式的确能加速模型在多轮对话上的训练,我们考虑在近期实现类似的功能,感谢你的建议!

期待!我lora微调实验,vicuna那种多轮语料构建方式,效果要好于prompt全遮。不知道qlora会不会有变化,估计也会好一些

from llama-factory.

flaviadeutsch avatar flaviadeutsch commented on May 18, 2024

期待+1

from llama-factory.

nieallen avatar nieallen commented on May 18, 2024

还有请问后续可以实现RWKV的lora微调吗?RWKV真的很快,感觉是gpt生成速度的两倍。但它不是纯transformers架构,不能用peft做lora训练,没有实现的脚本现在

from llama-factory.

hiyouga avatar hiyouga commented on May 18, 2024

在最新的代码 b6faf02 中,我们实现了多轮对话语料的训练。

另外,我们暂时不会考虑加入 RWKV 的微调。

from llama-factory.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.