Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Plan of PyTorch version? #1

Closed
RusticKey opened this issue Mar 2, 2021 · 3 comments
Closed

Plan of PyTorch version? #1

RusticKey opened this issue Mar 2, 2021 · 3 comments

Comments

@RusticKey
Copy link

Hello, your work is very interesting and the Tensorflow implementation is fine. However, I am wondering if you guys have any plans of porting it into PyTorch or already having plans of releasing it? Thanks!
@razvanmarinescu
Copy link
Owner

razvanmarinescu commented Mar 2, 2021 •
edited

Yes! NVIDIA just released StyleGAN-ADA in Pytorch, so we'll port straight to that! We have some deadlines coming up in the next 2-3 weeks, but we will work on it afterwards.

@RusticKey
Copy link
Author

RusticKey commented Mar 3, 2021 •
edited

I see, thank you for the response and good luck! I am looking forward for the PyTorch version since most of our lab's projects are written in PyTorch.

@razvanmarinescu
Copy link
Owner

We have just released the PyTorch version here: https://github.com/razvanmarinescu/brgm-pytorch

In the PyTorch version, we also switched to StyleGAN-ADA, and further added a Variational Inference extension for sampling from the posterior.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@razvanmarinescu @RusticKey