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