Pytorch Version

[SURABHI-GUPTA]

Is pytorch version available for this ?

[Luca96]

Hi @SURABHI-GUPTA,
unfortunately I have only developed the keras version for this project.

Maybe you can try yourself to migrate the code to pytorch, and sent a pull-request. 😉