def _load_from_h5(self, group, **kwargs): load_h5_saveables(group['param_updaters'], self.param_updaters, **kwargs) load_h5_saveables(group['momentum_updaters'], self.momentum_updaters, **kwargs) self.training_iter.load_from_h5(group['training_iter'], **kwargs) self.validation_iter.load_from_h5(group['validation_iter'], **kwargs) self.num_epochs_seen = group.attrs['num_epochs_seen']
def _load_from_h5(self, group, **kwargs): load_h5_saveables(group['shared_layers'], self.shared_layers, **kwargs) load_h5_saveables(group['id_layers'], self.id_layers, **kwargs) load_h5_saveables(group['cam_dir_layers'], self.cam_dir_layers, **kwargs)
def load_member(member_name): load_h5_saveables(group[member_name], model.__dict__[member_name])