def main():
optim = Optimizer(PATCH_SIZE, RANGE_OF_ONES, GRANULARITY_TH, ACC_LOSS)
optim.base_line_result()
optim.by_uniform_layers()
optim.by_uniform_filters()
optim.by_uniform_patches()
optim.by_max_granularity()
def get_baseline_rec(net_name, dataset_name, ps, init_acc):
rec_finder = RecordFinder(net_name, dataset_name, ps, ('*', '*'), '*', '*',
init_acc)
bs_line_fn = rec_finder.find_rec_filename(None, RecordType.BASELINE_REC)
if bs_line_fn is None:
optim = Optimizer(ps, (None, None), None, None)
optim.base_line_result()
bs_line_fn = rec_finder.find_rec_filename(None,
RecordType.BASELINE_REC)
if bs_line_fn is None:
print(
f' !!! Was not able to get baseline result for initial accuracy of {init_acc} !!!'
)
print(f' !!! Adjust TEST_SET_SIZE in Config.py !!!')
return bs_line_fn
return load_from_file(bs_line_fn, '')
def eval_baseline_and_runtimes(ps, ones_range, gran_th, patterns_idx=None):
optim = Optimizer(ps, ones_range, gran_th, 0, patterns_idx=patterns_idx)
optim.base_line_result()
optim.print_runtime_eval()
return optim.init_acc
