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
