Thanks for sharing. Since it's your own repository I'd recommend to append a "Show HN:" at the beginning.

I've also learned of Netron to visualize a network's architecture. That's neat way to get a grasp of a new model (here a screenshot: https://raw.githubusercontent.com/lutzroeder/netron/master/....)