How to run pytorch code

[yifanjiang19]

Hi, thanks for sharing your code,

I can only see the tf example code on README, how to launch the pytorch example code?

Thanks,