From 94e2fc2b16bf2c29955903c200451bd1073f4dad Mon Sep 17 00:00:00 2001 From: zjut Date: Sat, 26 Oct 2024 18:35:42 +0800 Subject: [PATCH] =?UTF-8?q?=E6=97=A5=E5=BF=97=E6=96=87=E4=BB=B6=E6=9B=B4?= =?UTF-8?q?=E6=96=B0=E5=92=8C=E6=B5=8B=E8=AF=95=E8=84=9A=E6=9C=AC=E6=89=A9?= =?UTF-8?q?=E5=B1=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在日志文件中添加了PFCFuse模型的详细配置信息，包括模型参数和训练 --- logs/log_20241009_120404.log | 1 + test_IVF.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/logs/log_20241009_120404.log b/logs/log_20241009_120404.log index e69de29..136ba21 100644 --- a/logs/log_20241009_120404.log +++ b/logs/log_20241009_120404.log @@ -0,0 +1 @@ +2.4.1+cu121 True Model: PFCFuse Number of epochs: 60 Epoch gap: 40 Learning rate: 0.0001 Weight decay: 0 Batch size: 1 GPU number: 0 Coefficient of MSE loss VF: 1.0 Coefficient of MSE loss IF: 1.0 Coefficient of RMI loss VF: 1.0 Coefficient of RMI loss IF: 1.0 Coefficient of Cosine loss VF: 1.0 Coefficient of Cosine loss IF: 1.0 Coefficient of Decomposition loss: 2.0 Coefficient of Total Variation loss: 5.0 Clip gradient norm value: 0.01 Optimization step: 20 Optimization gamma: 0.5 \ No newline at end of file diff --git a/test_IVF.py b/test_IVF.py index c402ae7..875c938 100644 --- a/test_IVF.py +++ b/test_IVF.py @@ -19,7 +19,7 @@ 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/whaiFusion10-08-16-20.pth" -for dataset_name in ["TNO","RoadScene"]: +for dataset_name in ["TNO","RoadScene","sar"]: print("\n"*2+"="*80) model_name="PFCFuse " print("The test result of "+dataset_name+' :')