diff --git a/logs/log_20241009_120404.log b/logs/log_20241009_120404.log index e69de29..136ba21 100644 --- a/logs/log_20241009_120404.log +++ b/logs/log_20241009_120404.log @@ -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 \ No newline at end of file diff --git a/test_IVF.py b/test_IVF.py index c402ae7..875c938 100644 --- a/test_IVF.py +++ b/test_IVF.py @@ -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+' :')