94e2fc2b16
- 在日志文件中添加了PFCFuse模型的详细配置信息,包括模型参数和训练
1 line
488 B
Plaintext
1 line
488 B
Plaintext
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
|