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))