Increase training epochs

pksX01 · pksX01 · commit 5c66e0ba1dbb · 2024-09-30T22:04:48.000+05:30
diff --git a/examples/generative/cyclegan.py b/examples/generative/cyclegan.py
@@ -618,7 +618,7 @@ def discriminator_loss_fn(real, fake):
 # 7 minutes on a single P100 backed machine.
 cycle_gan_model.fit(
     tf.data.Dataset.zip((train_horses, train_zebras)),
-    epochs=1,
+    epochs=90,
     callbacks=[plotter, model_checkpoint_callback],
 )
 

Original file line number	Diff line number	Diff line change
`@@ -618,7 +618,7 @@ def discriminator_loss_fn(real, fake):`
`618`	`618`	`# 7 minutes on a single P100 backed machine.`
`619`	`619`	`cycle_gan_model.fit(`
`620`	`620`	`tf.data.Dataset.zip((train_horses, train_zebras)),`
`621`		`- epochs=1,`
	`621`	`+ epochs=90,`
`622`	`622`	`callbacks=[plotter, model_checkpoint_callback],`
`623`	`623`	`)`
`624`	`624`