I think so - OP if you were to include a Jupyter notebook it would save some time
I second this
OP, just to balance out these comments, you hooked me with the readme. DNNs have been moving way faster than pedagogy can keep up and these sorts of learning experiments are super helpful.