Training code for pytorch?

Hi, great job! I have tested your pytorch code, but wondering if the training code will be available for pytorch? Thanks.

The repository is primarily in Lua/Torch7. To add PyTorch training code, first understand the existing model architecture and data pipeline in the Lua codebase. Review the paper and dataset to replicate the training process. This involves significant design and implementation work to port the training logic to PyTorch.

索引モデルが issue の本文から書いたものです。