Shortcuts

CoLLiE.controller

Controller

Trainer

CoLLie 训练器,支持快速分布式训练和验证。

EvaluatorForGeneration

EvaluatorForPerplexity

EvaluatorForClassfication

专为分类任务提供的检验器 :type only_latin: bool, default: True :param only_latin: 当使用 helm 风格(即基于生成的)分类检验时,是否去除拉丁字母和数字和空格以外的字符。 :type max_new_tokens: int, default: 3 :param max_new_tokens: 当使用 helm 风格(即基于生成的)分类检验时,生成的 token 数量。 一般为了提高准确率可以多生成几个 token。

其它版本