Open mmxuan18 opened 5 years ago
I haven't tested it out on pytorch 1.0 yet, but I imagine it should work! It'll probably take a bit of work to get it working with the JIT compiler
@gpleiss will you continue support the new jit feature?
I would like to, but I probably won't have time in the near future. Open to a PR though :)
I haven't tested it out on pytorch 1.0 yet, but I imagine it should work! It'll probably take a bit of work to get it working with the JIT compiler