I’m not familiar with the smp repository and their metrics, but based on the error message I would suggest to check the inputs to this particular metric function and check what the expected shapes would be (e.g. you might miss an argmax operation or any other op which changes the shape).