Skip to content

Commit 425db71

Browse files
author
skeydan
committed
re-add tf_function functionality
1 parent 78a4de6 commit 425db71

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

vignettes/examples/eager_dcgan.R

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,9 @@ discriminator <-
127127
generator <- generator()
128128
discriminator <- discriminator()
129129

130+
generator$call <- tf$`function`(generator$call)
131+
discriminator$call <- tf$`function`(discriminator$call)
132+
130133
cross_entropy = tf$keras$losses$BinaryCrossentropy(from_logits = TRUE)
131134

132135
discriminator_loss <- function(real_output, generated_output) {
@@ -142,7 +145,7 @@ generator_loss <- function(generated_output) {
142145
discriminator_optimizer <- tf$optimizers$Adam(1e-4)
143146
generator_optimizer <- tf$optimizers$Adam(1e-4)
144147

145-
num_epochs <- 150
148+
num_epochs <- 1#50
146149
noise_dim <- 100
147150
num_examples_to_generate <- 25L
148151

0 commit comments

Comments
 (0)