diff --git a/logs/log_20241006_success.log b/logs/log_20241006_success.log new file mode 100644 index 0000000..2ab40fd --- /dev/null +++ b/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