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