26 lines
1.0 KiB
Plaintext
26 lines
1.0 KiB
Plaintext
|
/home/star/anaconda3/envs/pfcfuse/bin/python /home/star/whaiDir/PFCFuse/train.py
|
||
|
|
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
|
||
|
|
[Epoch 39/60] [Batch 6486/6487] [loss: 0.002562] ETA: 3:30:05.95/home/star/whaiDir/PFCFuse/utils/loss.py:15: UserWarning: Using a target size (torch.Size([1, 1, 128, 128])) that is different to the input size (torch.Size([])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.
|
||
|
|
loss_rmi=F.l1_loss(x_rmi_max, generate_img)
|
||
|
|
[Epoch 59/60] [Batch 6486/6487] [loss: 2.106119] ETA: 0:00:00.08
|
||
|
|
Process finished with exit code 0
|