Is there a Pytorch version available?

[iamtonymwt]

Hi. This is a really impressive work. Is there a Pytorch version code available?

[tkipf]

Thanks a lot for your question! There is a work-in-progress re-implementation by @junkeun-yi at: https://github.com/junkeun-yi/SAVi-pytorch -- Note that this is not an official implementation of SAVi and we did not check the code.

You can also follow this GitHub Issue for more details: #3

[tkipf]

Re-opening for visibility.

[Genera1Z]

Hi. This is a really impressive work. Is there a Pytorch version code available?

@iamtonymwt I tried tkipf's recommendation and have spent a lot of time as well as effort on it, but it didn't work well. Then I myself implemented a pretty readable and usable SAVi of PyTorch version, which is 3x faster for training while achieving a bit better performance. Please find it here: https://github.com/Genera1Z/SAVi-PyTorch