def __init__(self):
"""Constructor function."""
# class-independent initialization
super(ModelHelper, self).__init__()
# initialize training & evaluation subsets
self.dataset_train = Ilsvrc12Dataset(is_train=True)
self.dataset_eval = Ilsvrc12Dataset(is_train=False)
def __init__(self, data_format='channels_last'):
"""Constructor function."""
# class-independent initialization
assert data_format == 'channels_last', 'MobileNet only supports \'channels_last\' data format'
super(ModelHelper, self).__init__(data_format)
# initialize training & evaluation subsets
self.dataset_train = Ilsvrc12Dataset(is_train=True)
self.dataset_eval = Ilsvrc12Dataset(is_train=False)
def __init__(self, data_path, netcfg):
self.data_scope = 'DATA'
self.model_scope = 'HRNET'
# initialize training & evaluation subsets
self.dataset_train = Ilsvrc12Dataset(is_train=True, data_dir=data_path)
self.dataset_eval = Ilsvrc12Dataset(is_train=False, data_dir=data_path)
# initialize network
self.hrnet = HRNet(netcfg)
# learning rate
self.lr_init = self.hrnet.cfg['COMMON']['lr_rate_init']
self.model_path = './models'
self.log_path = './logs'
self.summ_step = self.hrnet.cfg['COMMON']['summary_step']
self.save_step = self.hrnet.cfg['COMMON']['save_step']
self.nb_iters_start = 0