test: 更新测试模型路径

- 修改了 test_IVF.py 文件中的模型路径
- 将旧路径 whaiFusion11-15-22-09.pth 更改为新路径 whaiFusion11-16-11-20.pth
This commit is contained in:
zjut 2024-11-16 21:37:02 +08:00
parent e8a0212bbb
commit 9224f9b640

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"
ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-15-22-09.pth" ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-16-11-20.pth"
for dataset_name in ["sar"]: for dataset_name in ["sar"]:
print("\n"*2+"="*80) print("\n"*2+"="*80)