Hm, I can actually try the training on my GPU. One of the things I want to try next. Maybe a bit more complex than a fish :)