Comment by darepublic

Amazing, I was looking to do something very similar for myself-- dig into pytorch library to see how everything was working.