Skip to content

设置remove_unused_columns为false,在gkd的compute_loss函数中仍然无法透传额外字段 #4923

Open
@YihengLiu1996

Description

@YihengLiu1996

调试发现,在encode输出的res中含有extra kwargs字段,但在进入到data_collator函数后,batch中就没有extra kwargs字段了。

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions