Hi everyone. I’m passing custom compute metric to the trainer. However I need to provide an additional argument besides the batch. How do I do that? My custom compute metric function: def compute_metrics(p, label_list): ... My trainer: trainer = Trainer( …

Pass Arguments to custom compute_metrics in Trainer

uygarkurt April 10, 2023, 1:39pm 3

thank you

Topic		Replies	Views
Custom model for Trainer 🤗Transformers	1	381	July 8, 2023
Custom metrics with extra data? Beginners	8	3541	April 12, 2024
How to define the compute_metrics() function in Trainer? 🤗Transformers	3	16249	December 20, 2021
How to add a custom argument to TrainingArguments? 🤗Transformers	2	4594	June 5, 2022
Trainer doesn't call compute_metrics during evaluation 🤗Transformers	2	828	June 24, 2024