Zhang Jiarui

icon company Tongji University icon location Shanghai icon location Zhang Jiarui, Tongji University, Data Science

Results 1 comments of Zhang Jiarui

added warning to Trainer when label_names is not specified for PeftModel

> Basically inside Trainer's init function, it automatically finds whether 'labels' argument exist in model class's forward function. > > e.g. LlamaForCausalLM: have labels field in forward, so 'labels' is...