Skip to content

Commit 39c2f04

Browse files
authored
Update eager_dcgan.R
1 parent 9d63d99 commit 39c2f04

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

vignettes/examples/eager_dcgan.R

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -195,15 +195,15 @@ train <- function(dataset, epochs, noise_dim) {
195195
})
196196

197197
gradients_of_generator <-
198-
gen_tape$gradient(gen_loss, generator$variables)
198+
gen_tape$gradient(gen_loss, generator$trainable_variables)
199199
gradients_of_discriminator <-
200-
disc_tape$gradient(disc_loss, discriminator$variables)
200+
disc_tape$gradient(disc_loss, discriminator$trainable_variables)
201201

202202
generator_optimizer$apply_gradients(purrr::transpose(list(
203-
gradients_of_generator, generator$variables
203+
gradients_of_generator, generator$trainable_variables
204204
)))
205205
discriminator_optimizer$apply_gradients(purrr::transpose(
206-
list(gradients_of_discriminator, discriminator$variables)
206+
list(gradients_of_discriminator, discriminator$trainable_variables)
207207
))
208208

209209
total_loss_gen <- total_loss_gen + gen_loss

0 commit comments

Comments
 (0)