8bf6fdc8160d8e6150a0b7211b5ee41496331057,eval_ensemble.py,,,#,22

Before Change

opt = parser.parse_args()

model_infos = [utils.pickle_load(open("log_%s/infos_%s-best.pkl" %(id, id))) for id in opt.ids]
model_paths = ["log_%s/model-best.pth" %(id) for id in opt.ids]

// Load one infos
infos = model_infos[0]

After Change

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

// Load one infos
infos = model_infos[0]

// override and collect parameters
replace = ["input_fc_dir", "input_att_dir", "input_box_dir", "input_label_h5", "input_json", "batch_size", "id"]

Instances