Support PyTorch 0.4.0 #10
Description
Migrate existing code to work in PyTorch version 0.4.0. PyTorch 0.4.0 introduced a major core changes that changed the previous API version.
Status: I have started work on this and making code changes by referring to the official PyTorch migration guide.