-
pytorch-pretrained-BigGAN
🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts.
To my surprise, there don't seem to be many GAN weights available for download. Worse yet, many that are available (e.g., https://github.com/huggingface/pytorch-pretrained-BigGAN ) only come with pre-trained generator weights, not discriminator weights. But I need both.
-
CodeRabbit
CodeRabbit: AI Code Reviews for Developers. Revolutionize your code reviews with AI. CodeRabbit offers PR summaries, code walkthroughs, 1-click suggestions, and AST-based analysis. Boost productivity and code quality across all major languages with each PR.
-
The second link gives you the entire source code for training the model https://github.com/ajbrock/BigGAN-PyTorch/tree/master . Looks like BigGAN.py and BigGANdeep.py are the two files that define the architecture. Can you work with that?