Custom model for Trainer

Hello,
trying to figure out everything needed when I train a custom model with trainer.

  1. As far as I understand - output should be tuple with loss value first.
  2. the function provided to compute_metrics takes EvalPrediction. How is the value passed to my compute metrics function created? How does trainer know how to build the input to this compute_metrics function

Thanks!

1 Like

Up. Any answer for this? :blush: