Thanks lolongcovas, im just going to transfer it to caffe2 and run it there for production.
Also, found this nice tutorial, so I guess the pytorch guys know my needs before I do 
http://pytorch.org/tutorials/advanced/super_resolution_with_caffe2.html 212