Please take more care of your repository.
Plenty of PRs unmerged. Suggestions unaddressed. It's not clear which ones to implement and which ones to ignore.
Initial code doesn't run.
After tweaking:
train accuracy: 0.07
train accuracy: 0.17
train accuracy: 0.14
train accuracy: 0.07
train accuracy: 0.07
I don't see how you get the results your notebook claims.
This has proven to be a headache and a frustration. Clearly so much effort and intelligence was put into the construction, why not put in the extra 1% and leave behind a valuable/tidy/clean resource?