How to integrate the pytorch version

[alextaymx]

Hi Author,
Your work is awesome, I wish I could use it in my university project. May I know how to integrate the pytorch version using our own dataset? I'm currently working on rp2k_dataset which is mainly for product classification.

Also , which version of pytorch or any dependencies required to run your work ya?

Any help is greatly appreciated, thanks.

Regards ,
Alex