Summaries for both models can be found under the following links, just in case.
https://gist.github.com/Nbooo/7d3f6f5c4dc8768f765b02408b66e6d0 2 (pytorch)
https://gist.github.com/Nbooo/2b47b384c20e2dfb5e1224083b2ca2f3 (keras)