search

dorarad / gansformer

Generative Adversarial Transformers
MIT License
1.32k stars 149 forks source link

Pytorch version #35

Closed di-mi-ta closed 2 years ago

di-mi-ta commented 2 years ago

Hi,

This work is great! I have a question that does the PyTorch version has full features (training, inference on all datasets) like the Tensorflow version?

Thanks!

dorarad commented 2 years ago

Hi, yep the PyTorch version has training and inference on all datasets like the TF version! The only thing not in the PyTorch version are couple of baseline models (k-GAN and SAGAN).