https://github.com/dtsip/in-context-learning/blob/338337192195827c58356f9b627e5f999ea703d2/src/eval.py#L60 I think this line should be metrics[:, i] = task.get_metric()(pred.cpu(), ys[:, i])
in-context-learning/src/eval.py
Line 60 in 3383371
I think this line should be metrics[:, i] = task.get_metric()(pred.cpu(), ys[:, i])