Fixed hardcoded number of epochs

vinayak19th · vinayak19th · commit 7544abbda629 · 2025-10-30T22:59:06.000-07:00
diff --git a/examples/vision/nl_image_search.py b/examples/vision/nl_image_search.py
@@ -574,7 +574,7 @@ def test_step(self, features):
 
 history = dual_encoder.fit(
     train_dataset,
-    epochs=5,
+    epochs=num_epochs,
     validation_data=valid_dataset,
     callbacks=[reduce_lr, model_checkpoint_callback],
 )

Original file line number	Diff line number	Diff line change
`@@ -574,7 +574,7 @@ def test_step(self, features):`
`574`	`574`
`575`	`575`	`history = dual_encoder.fit(`
`576`	`576`	`train_dataset,`
`577`		`- epochs=5,`
	`577`	`+ epochs=num_epochs,`
`578`	`578`	`validation_data=valid_dataset,`
`579`	`579`	`callbacks=[reduce_lr, model_checkpoint_callback],`
`580`	`580`	`)`