日志文件更新和测试脚本扩展
- 在日志文件中添加了PFCFuse模型的详细配置信息,包括模型参数和训练
This commit is contained in:
parent
0e6064181a
commit
94e2fc2b16
@ -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
|
@ -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+' :')
|
||||
|
Loading…
Reference in New Issue
Block a user