tf.keras中关于model.trainable=False的设置(in GAN)

提出问题

在看GAN的实现代码的时候,发现了这么一个地方:

class GAN():
    def __init__(self):
        self.img_rows = 28
        self.img_cols = 28
        self.channels = 1
        self.img_shape = (self.img_rows, self.img_cols, self.channels)
        self.latent_dim = 100
​
        optimizer = Adam(0.0002, 0.5)
​
        # Build and compile the discriminator
        self.discriminator = self.build_discriminator()
        self.discriminator.compile(loss='binary_crossentropy',
            optimizer=optimizer,
            metrics=['accuracy'])
​
        # Build the generator
        self.generator = self.build_generator()
​
        # The generator takes noise as input and generates imgs
        z = Input(shape=(self.latent_dim,))
        img = self.generator(z)
​
        # For the combined model we will only train the generator
        self.discriminator.trainable = False
​
        # The discriminator takes generated images as input and determines validity
        validity = self.discriminator(img)
​
        # The combined model  (stacked generator and discriminator)
        # Trains the generator to fool the discriminator
        self.combined = Model(z, validity)
        self.combined.compile(loss='binary_crossentropy', optimizer=optimizer)

缩小上面代码的范围,看这一行:

 # For the combined model we will only train the generator
        self.discriminator
def define_generator(): # 定义输入 inputs = layers.Input(shape=(LATENT_DIM,)) x = layers.Dense(256)(inputs) x = layers.LeakyReLU()(x) x = layers.BatchNormalization()(x) x = layers.Dense(512)(x) x = layers.LeakyReLU()(x) x = layers.BatchNormalization()(x) x = layers.Dense(SEQ_LEN * NUM_CLASSES, activation='tanh')(x) outputs = layers.Reshape((SEQ_LEN, NUM_CLASSES))(x) # 定义模型 model = tf.keras.Model(inputs, outputs, name='generator') return model # 定义判别器模型 def define_discriminator(): # 定义输入 inputs = layers.Input(shape=(SEQ_LEN, NUM_CLASSES)) x = layers.Flatten()(inputs) x = layers.Dense(512)(x) x = layers.LeakyReLU()(x) x = layers.Dense(256)(x) x = layers.LeakyReLU()(x) # 注意这里输出为1,表示真假 outputs = layers.Dense(1, activation='sigmoid')(x) # 定义模型 model = tf.keras.Model(inputs, outputs, name='discriminator') return model # 定义GAN模型 def define_gan(generator, discriminator): # 将判别器设置为不可训练 discriminator.trainable = False # 定义输入 inputs = layers.Input(shape=(LATENT_DIM,)) # 生成音符和和弦 outputs = generator(inputs) # 判断音符和和弦是否为真实的 real_or_fake = discriminator(outputs) # 定义模型 model = tf.keras.Model(inputs, real_or_fake, name='gan') return model # 定义损失函数和优化器 def define_loss_and_optimizer(): loss_fn = tf.keras.losses.BinaryCrossentropy() generator_optimizer = tf.keras.optimizers.Adam(learning_rate=0.0002, beta_1=0.5) discriminator_optimizer = tf.keras.optimizers.Adam(learning_rate=0.0002, beta_1=0.5) return loss_fn, generator_optimizer, discriminator_optimizer
最新发布
06-03
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值