https://analyticsindiamag.com/tensorflow-vs-pytorch-which-framework-is-better-for-implementing-deep-learning-models/