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])
