ensure model initialized on ANY trackable attr set

This commit is contained in:
Dominic Jack 2020-06-05 08:54:38 +10:00
parent ec7e854ca9
commit 17b7e16913

View File

@ -319,7 +319,7 @@ class Model(base_layer.Layer, version_utils.ModelVersionSelector):
super(Model, self).__setattr__(name, value)
return
if all(
if any(
isinstance(v, (base_layer.Layer,
data_structures.TrackableDataStructure)) or
trackable_layer_utils.has_weights(v) for v in nest.flatten(value)):