diff --git a/.idea/PFCFuse.iml b/.idea/PFCFuse.iml
index a8d74d0..8b8c395 100644
--- a/.idea/PFCFuse.iml
+++ b/.idea/PFCFuse.iml
@@ -2,7 +2,7 @@
 <module type="PYTHON_MODULE" version="4">
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$" />
-    <orderEntry type="jdk" jdkName="Remote Python 3.8.10 (sftp://star@192.168.50.108:22/home/star/anaconda3/envs/pfcfuse/bin/python)" jdkType="Python SDK" />
+    <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
   <component name="PyDocumentationSettings">
diff --git a/.idea/misc.xml b/.idea/misc.xml
index d3caf2e..1586a88 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -12,5 +12,5 @@
       </MavenGeneralSettings>
     </option>
   </component>
-  <component name="ProjectRootManager" version="2" project-jdk-name="Remote Python 3.8.10 (sftp://star@192.168.50.108:22/home/star/anaconda3/envs/pfcfuse/bin/python)" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8 (pfcfuse)" project-jdk-type="Python SDK" />
 </project>
\ No newline at end of file
diff --git a/test_IVF.py b/test_IVF.py
index ef78c63..472c4cd 100644
--- a/test_IVF.py
+++ b/test_IVF.py
@@ -17,14 +17,16 @@ current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
 
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
-ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-17-10-34.pth"
+pth_path = "whaiFusion11-17-20-18"
+ckpt_path= r"/home/star/whaiDir/PFCFuse/models/"+pth_path+".pth"
+print("path_pth:{}".format(ckpt_path))
 
 for dataset_name in ["sar"]:
     print("\n"*2+"="*80)
-    model_name="PFCFuse Enhance  增加widthblock"
+    model_name=pth_path
     print("The test result of "+dataset_name+' :')
     test_folder = os.path.join('test_img', dataset_name)
-    test_out_folder=os.path.join('test_result',current_time,dataset_name)
+    test_out_folder=os.path.join('test_result',pth_path,dataset_name)
 
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
     Encoder = nn.DataParallel(Restormer_Encoder()).to(device)