model = BaseModel(opt) L1s = [] SSIMs = [] with torch.no_grad(): for idx, data in enumerate(tqdm(dataset)): ida = data['id_a'][0].split('_') idb = data['id_b'][0].split('_') assert (ida[0] == idb[0]) model_id = ida[0] ida = '_'.join(ida[1:]) idb = '_'.join(idb[1:]) model.set_input(data) model.switch_mode('eval') model.anim_dict = {'vis': []} model.real_A = model.real_A[:1] model.real_B = model.real_B[:1] eval_res = model.evaluate() L1s.append(eval_res['L1']) SSIMs.append(eval_res['SSIM']) print('L1:{l1}, SSIM:{ssim}'.format(l1=np.mean(L1s), ssim=np.mean(SSIMs)))