More TF fixes (#28081)

* More build_in_name_scope()

* Make sure we set the save spec now we don't do it with dummies anymore

* make fixup
This commit is contained in:
Matt
2023-12-18 15:26:03 +00:00
committed by GitHub
parent 0695b2421a
commit 71d47f0ad4
5 changed files with 6 additions and 5 deletions

View File

@@ -217,7 +217,7 @@ class TFCoreModelTesterMixin:
for model_class in self.all_model_classes[:2]:
class_inputs_dict = self._prepare_for_class(inputs_dict, model_class)
model = model_class(config)
model.build()
model.build_in_name_scope()
num_out = len(model(class_inputs_dict))
for key in list(class_inputs_dict.keys()):