--

logs/log_20241006_success.log

@@ -0,0 +1,25 @@
+/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