We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6885b3b commit 69ce357Copy full SHA for 69ce357
tensorflow_recommenders_addons/dynamic_embedding/python/keras/models.py
@@ -124,7 +124,7 @@ def _traverse_emb_layers_and_save(proc_size=1, proc_rank=0):
124
# save optimizer parameters of Dynamic Embedding
125
if include_optimizer is True:
126
de_opt_vars = a2a_emb.optimizer_vars.as_list() if hasattr(
127
- a2a_emb.optimizer_vars, "as_list") else a2a_emb.optimizer_vars
+ a2a_emb.optimizer_vars, "as_list") else a2a_emb.optimizer_vars
128
for de_opt_var in de_opt_vars:
129
de_opt_var.save_to_file_system(dirpath=de_dir,
130
proc_size=proc_size,
0 commit comments