import tensorflow as tf

            start_resolutions=(4, 4), scale_base=2, num_resolutions=3))
    fake_loss = tf.reduce_sum(tf.square(x))
    grad_norms = [
        _get_grad_norm(
            fake_loss, tf.trainable_variables('.*/progressive_gan_block_1/.*')),
        _get_grad_norm(
            fake_loss, tf.trainable_variables('.*/progressive_gan_block_2/.*')),
        _get_grad_norm(
