Update for Pytorch 1.0+?

[WonderSeven]

Hi, this is a nice work and the code is easy to reuse, do you have any scheduler to update your code for a new pytorch verison (like 1.0+)? I am really aware of this reposity is based on pytorch-spn, which is implemented in pytorch 0.4, but pytorch 0.4 shows poor compatibility with later version.

[ayaz0201]

Hi,

Could you please port to pytorch 1.0