def _save_to_h5(self, group): save_h5_saveables(self.param_updaters, group.create_group('param_updaters')) save_h5_saveables(self.momentum_updaters, group.create_group('momentum_updaters')) self.training_iter.save_to_h5(group.create_group('training_iter')) self.validation_iter.save_to_h5(group.create_group('validation_iter')) group.attrs['num_epochs_seen'] = self.num_epochs_seen
def _save_to_h5(self, group): save_h5_saveables(self.shared_layers, group.create_group('shared_layers')) save_h5_saveables(self.id_layers, group.create_group('id_layers')) save_h5_saveables(self.cam_dir_layers, group.create_group('cam_dir_layers'))
def save_member(member_name): h5_saveables = self.__dict__[member_name] save_h5_saveables(h5_saveables, group.create_group(member_name))