日志文件更新和测试脚本扩展

- 在日志文件中添加了PFCFuse模型的详细配置信息,包括模型参数和训练
This commit is contained in:
zjut 2024-10-26 18:35:42 +08:00
parent 0e6064181a
commit 94e2fc2b16
2 changed files with 2 additions and 1 deletions

View File

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

View File

@ -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+' :')