# Input arguments and options parser = argparse.ArgumentParser() # Input paths parser.add_argument('--ids', nargs='+', required=True, help='id of the models to ensemble') parser.add_argument('--weights', nargs='+', required=False, default=None, help='id of the models to ensemble') # parser.add_argument('--models', nargs='+', required=True # help='path to model to evaluate') # parser.add_argument('--infos_paths', nargs='+', required=True, help='path to infos to evaluate') opts.add_eval_options(parser) opts.add_diversity_opts(parser) opt = parser.parse_args() model_infos = [] model_paths = [] for id in opt.ids: if '-' in id: id, app = id.split('-') app = '-' + app else: app = '' model_infos.append( utils.pickle_load(open('log_%s/infos_%s%s.pkl' % (id, id, app)))) model_paths.append('log_%s/model%s.pth' % (id, app))
def add_eval_options(): parser = argparse.ArgumentParser() opts.add_eval_options(parser) args = parser.parse_args() print('\n* add_eval_options', vars(args))