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.