Disable keras optimizer module change in SOK with use_legacy_optimizer (#1064)

oliverholworthy · jperez999 · commit a15b7c42c483 · 2023-04-24T14:14:36.000-04:00
diff --git a/merlin/models/tf/distributed/backend.py b/merlin/models/tf/distributed/backend.py
@@ -30,4 +30,4 @@
 
 
 if sok_installed:
-    sok.init()
+    sok.init(use_legacy_optimizer=False)

Original file line number	Diff line number	Diff line change
`@@ -30,4 +30,4 @@`
`30`	`30`
`31`	`31`
`32`	`32`	`if sok_installed:`
`33`		`- sok.init()`
	`33`	`+ sok.init(use_legacy_optimizer=False)`