a258cacef0569253614063bfb611d7c2b22f19bd,eval_ensemble.py,,,#,22

Before Change

    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)
model.seq_length = opt.seq_length

After Change

    _models.append(tmp)

if opt.weights:
    opt.weights = [float(_) for _ in opt.weights]
model = AttEnsemble(_models, weights=opt.weights)
model.seq_length = opt.seq_length
model.cuda()
model.eval()

Instances