How to integrate the pytorch version

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