About the StyleGAN

mirak · September 7, 2021, 11:19am

Hi,

I just finished the assignment for course 2 week 3, and I would like to know if someone can provide me with the full implementation of the StyleGAN (the training loop),

thanks in advance,

Khushwanth · September 7, 2021, 11:49am

github.com

NVlabs/stylegan/blob/master/training/training_loop.py

# Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
#
# This work is licensed under the Creative Commons Attribution-NonCommercial
# 4.0 International License. To view a copy of this license, visit
# http://creativecommons.org/licenses/by-nc/4.0/ or send a letter to
# Creative Commons, PO Box 1866, Mountain View, CA 94042, USA.

"""Main training script."""

import os
import numpy as np
import tensorflow as tf
import dnnlib
import dnnlib.tflib as tflib
from dnnlib.tflib.autosummary import autosummary

import config
import train
from training import dataset
from training import misc

This file has been truncated. show original

here is the training loop that was published in the paper

aray · September 8, 2021, 12:57am

Here is a very simple StyleGAN2 implementation with a lot of comments:

fzhurd · September 8, 2021, 3:16am

Here is another simple StyleGAN with PyTorch example

Topic		Replies	Views
Need help in debugging the stylegan code Build Better Generative Adversarial Networks week-module-3	4	545	March 16, 2023
Issue with finetuning style gan2 Build Better Generative Adversarial Networks week-module-3 , coursera-platform	2	29	November 15, 2025
StyleGAN not training Apply Generative Adversarial Networks week-module-3	1	555	July 26, 2023
Art Generation with Neural Style Transfef Convolutional Neural Networks coursera-platform	3	951	March 1, 2022
Art _Generation with Neural Style Transfer Convolutional Neural Networks coursera-platform	1	642	August 6, 2021

About the StyleGAN

Related topics