torch.zeros_like(outpde_real)有问题吗
时间: 2024-05-30 15:11:26 浏览: 13
torch.zeros_like(outpde_real)的作用是创建一个与outpde_real形状相同的张量,并将其所有元素设置为0。如果outpde_real是一个张量,则这个函数没有问题,它会返回一个相同形状的全零张量。但是,如果outpde_real不是一个张量,那么这个函数就会出错。因此,需要确保outpde_real是一个张量,才能安全地使用torch.zeros_like(outpde_real)。
相关问题
for i, (x_test, c_test) in enumerate(test_dataloader): _, _, _ = vae(x_test, c_test) real_y = gan(vae.latent) z = torch.rand_like(vae.latent) fake_y = gan(z) gan_real_loss = gan_criterion(real_y, torch.ones_like(real_y)) gan_fake_loss = gan_criterion(fake_y, torch.zeros_like(fake_y)) real_score = 1-gan_real_loss.mean().detach() fake_score = gan_fake_loss.mean().detach() real_score_mean.append(real_score.numpy()) fake_score_mean.append(fake_score.numpy())
这是一个使用GAN评估VAE生成样本质量的代码段。代码中首先从测试集中读取图像和标签,然后将它们输入到VAE模型中进行编码解码,得到重构图像和潜在变量。接着,将潜在变量输入到已经训练好的GAN模型中,得到GAN的判别结果real_y和fake_y。之后,使用GAN的损失函数gan_criterion分别计算real_y和fake_y的损失gan_real_loss和gan_fake_loss。接着,通过计算real_y和fake_y的平均值,得到它们对应的真实分数real_score和虚假分数fake_score。最后,将real_score和fake_score的值分别添加到real_score_mean和fake_score_mean列表中,用于计算整个测试集上GAN的真实分数和虚假分数的平均值。这个代码段的目的是为了通过GAN的真实分数和虚假分数来评估VAE生成样本的质量。
def train_step(real_ecg, dim): noise = tf.random.normal(dim) for i in range(disc_steps): with tf.GradientTape() as disc_tape: generated_ecg = generator(noise, training=True) real_output = discriminator(real_ecg, training=True) fake_output = discriminator(generated_ecg, training=True) disc_loss = discriminator_loss(real_output, fake_output) gradients_of_discriminator = disc_tape.gradient(disc_loss, discriminator.trainable_variables) discriminator_optimizer.apply_gradients(zip(gradients_of_discriminator, discriminator.trainable_variables)) ### for tensorboard ### disc_losses.update_state(disc_loss) fake_disc_accuracy.update_state(tf.zeros_like(fake_output), fake_output) real_disc_accuracy.update_state(tf.ones_like(real_output), real_output) ####################### with tf.GradientTape() as gen_tape: generated_ecg = generator(noise, training=True) fake_output = discriminator(generated_ecg, training=True) gen_loss = generator_loss(fake_output) gradients_of_generator = gen_tape.gradient(gen_loss, generator.trainable_variables) generator_optimizer.apply_gradients(zip(gradients_of_generator, generator.trainable_variables)) ### for tensorboard ### gen_losses.update_state(gen_loss) ####################### def train(dataset, epochs, dim): for epoch in tqdm(range(epochs)): for batch in dataset: train_step(batch, dim) disc_losses_list.append(disc_losses.result().numpy()) gen_losses_list.append(gen_losses.result().numpy()) fake_disc_accuracy_list.append(fake_disc_accuracy.result().numpy()) real_disc_accuracy_list.append(real_disc_accuracy.result().numpy()) ### for tensorboard ### # with disc_summary_writer.as_default(): # tf.summary.scalar('loss', disc_losses.result(), step=epoch) # tf.summary.scalar('fake_accuracy', fake_disc_accuracy.result(), step=epoch) # tf.summary.scalar('real_accuracy', real_disc_accuracy.result(), step=epoch) # with gen_summary_writer.as_default(): # tf.summary.scalar('loss', gen_losses.result(), step=epoch) disc_losses.reset_states() gen_losses.reset_states() fake_disc_accuracy.reset_states() real_disc_accuracy.reset_states() ####################### # Save the model every 5 epochs # if (epoch + 1) % 5 == 0: # generate_and_save_ecg(generator, epochs, seed, False) # checkpoint.save(file_prefix = checkpoint_prefix) # Generate after the final epoch display.clear_output(wait=True) generate_and_save_ecg(generator, epochs, seed, False)
下面是将上述代码转换为PyTorch代码的示例:
```python
def train_step(real_ecg, dim):
noise = torch.randn(dim)
for i in range(disc_steps):
generated_ecg = generator(noise, training=True)
real_output = discriminator(real_ecg, training=True)
fake_output = discriminator(generated_ecg, training=True)
disc_loss = discriminator_loss(real_output, fake_output)
discriminator.zero_grad()
disc_loss.backward()
discriminator_optimizer.step()
### for tensorboard ###
disc_losses.update(disc_loss)
fake_disc_accuracy.update(torch.zeros_like(fake_output), fake_output)
real_disc_accuracy.update(torch.ones_like(real_output), real_output)
#######################
for i in range(gen_steps):
generated_ecg = generator(noise, training=True)
fake_output = discriminator(generated_ecg, training=True)
gen_loss = generator_loss(fake_output)
generator.zero_grad()
gen_loss.backward()
generator_optimizer.step()
### for tensorboard ###
gen_losses.update(gen_loss)
#######################
def train(dataset, epochs, dim):
for epoch in tqdm(range(epochs)):
for batch in dataset:
train_step(batch, dim)
disc_losses_list.append(disc_losses.avg)
gen_losses_list.append(gen_losses.avg)
fake_disc_accuracy_list.append(fake_disc_accuracy.avg)
real_disc_accuracy_list.append(real_disc_accuracy.avg)
### for tensorboard ###
# with disc_summary_writer.as_default():
# tf.summary.scalar('loss', disc_losses.result(), step=epoch)
# tf.summary.scalar('fake_accuracy', fake_disc_accuracy.result(), step=epoch)
# tf.summary.scalar('real_accuracy', real_disc_accuracy.result(), step=epoch)
# with gen_summary_writer.as_default():
# tf.summary.scalar('loss', gen_losses.result(), step=epoch)
#######################
disc_losses.reset()
gen_losses.reset()
fake_disc_accuracy.reset()
real_disc_accuracy.reset()
#######################
# Save the model every 5 epochs
# if (epoch + 1) % 5 == 0:
# generate_and_save_ecg(generator, epochs, seed, False)
# checkpoint.save(file_prefix = checkpoint_prefix)
# Generate after the final epoch
# display.clear_output(wait=True)
# generate_and_save_ecg(generator, epochs, seed, False)
```
注意:上述代码仅作为示例,可能需要根据实际情况进行调整和修改。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)