a258cacef0569253614063bfb611d7c2b22f19bd,eval_ensemble.py,,,#,22

Before Change

    model_infos[i]["opt"].start_from = None
    tmp = models.setup(model_infos[i]["opt"])
    tmp.load_state_dict(torch.load(model_paths[i]))
    tmp.cuda()
    tmp.eval()
    _models.append(tmp)

model = AttEnsemble(_models)

After Change

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=True, 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")
// Basic options

Instances