whaifree/pfcfuse
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(test_IVF): 更新模型路径变量

Browse Source 
- 将 pth_path 变量的值从 "whaiFusion11-17-20-18" 修改为 "whaiFusion11-18-10-04"
- 此修改更新了测试脚本中使用的模型版本
This commit is contained in:
zjut 2024-11-18 15:41:14 +08:00
parent f28bb255f4
commit a218392d1c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test_IVF.py
View File

@ -17,7 +17,7 @@ current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
os.environ["CUDA_VISIBLE_DEVICES"] = "0" 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" ckpt_path= r"/home/star/whaiDir/PFCFuse/models/"+pth_path+".pth"
print("path_pth:{}".format(ckpt_path)) print("path_pth:{}".format(ckpt_path))