From a218392d1c85c06117cdb8df3dde8e340dac1614 Mon Sep 17 00:00:00 2001 From: zjut Date: Mon, 18 Nov 2024 15:41:14 +0800 Subject: [PATCH] =?UTF-8?q?refactor(test=5FIVF):=20=E6=9B=B4=E6=96=B0?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E8=B7=AF=E5=BE=84=E5=8F=98=E9=87=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 将 pth_path 变量的值从 "whaiFusion11-17-20-18" 修改为 "whaiFusion11-18-10-04" - 此修改更新了测试脚本中使用的模型版本 --- test_IVF.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test_IVF.py b/test_IVF.py index 472c4cd..b7e11c1 100644 --- a/test_IVF.py +++ b/test_IVF.py @@ -17,7 +17,7 @@ current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") os.environ["CUDA_VISIBLE_DEVICES"] = "0" -pth_path = "whaiFusion11-17-20-18" +pth_path = "whaiFusion11-18-10-04" ckpt_path= r"/home/star/whaiDir/PFCFuse/models/"+pth_path+".pth" print("path_pth:{}".format(ckpt_path))