def save(self, parent_saver: Saver):
saver = parent_saver.create_child(self._unit_folder_name)
self._save(saver)
# We need this to be created now so that we can store the tensors inside.
folder_path = saver.get_full_folder_path()
saver.description['tensors'] = self.save_tensors(folder_path)
def save(self,
parent_saver: Saver,
persistence_name: Optional[str] = None):
"""Save the node in the context of its parent saver."""
folder_name = persistence_name or self._get_persistence_name()
saver = parent_saver.create_child(folder_name)
self._save(saver)
def _save(self, saver: Saver):
super()._save(saver)
for i, unit in enumerate(self._units):
unit.save(saver.create_child(f'sub_unit_{i}'))
