HenryT:

IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)


Given this error I would think something in your code want some input size to be [N, M] and not just [N,]
Your y tensor in shape just [455]. You may try make it [455,1]