Pytorch>1.01

Hello. Thanks for your wonderful work! I wanted to use this optical flow network, but am constrained to have Pytorch>1.3. When i try to run `python setup.py install` i get errors. Do you support higher Pytorch versions?