How can I make a custom metric that return more than 1 number?

You’re welcome!

Passing lists is a bit tricky. See my earlier comment about this in another thread:

The change is in model.py:

    def step(self, xs, y, epoch):
        xtra = []
        output = self.m(*xs)

becomes:

    def step(self, xs, y, epoch):
        xtra = []
        if len(xs)>1 :output = self.m([*xs])
        else: output = self.m(*xs)

And the same change in the evaluate method as well.