The code saves the best testing results on Image-Text Retrieval

[yangbang18]

Hi, the following code is inappropriate:

X-VLM/Retrieval.py

Lines 310 to 333 in e7b9602

	val_result = itm_eval(score_val_i2t, score_val_t2i, val_loader.dataset.txt2img, val_loader.dataset.img2txt)
	print(val_result)
	test_result = itm_eval(score_test_i2t, score_test_t2i, test_loader.dataset.txt2img, test_loader.dataset.img2txt)
	print(test_result)
	log_stats = {**{f'train_{k}': v for k, v in train_stats.items()},
	**{f'val_{k}': v for k, v in val_result.items()},
	**{f'test_{k}': v for k, v in test_result.items()},
	'epoch': epoch}
	with open(os.path.join(args.output_dir, "log.txt"), "a") as f:
	f.write(json.dumps(log_stats) + "\n")
	if test_result['r_mean'] > best:
	save_obj = {
	'model': model_without_ddp.state_dict(),
	# 'optimizer': optimizer.state_dict(),
	# 'lr_scheduler': lr_scheduler.state_dict(),
	'config': config,
	# 'epoch': epoch,
	}
	torch.save(save_obj, os.path.join(args.output_dir, 'checkpoint_best.pth'))
	best = test_result['r_mean']
	best_epoch = epoch