add tau

checkpoints/ROMA_UNSB_001/loss_log.txt

@@ -0,0 +1,2 @@
+================ Training Loss (Sun Feb 23 15:46:44 2025) ================
+================ Training Loss (Sun Feb 23 15:52:29 2025) ================

checkpoints/ROMA_UNSB_001/train_opt.txt

@@ -0,0 +1,82 @@
+----------------- Options ---------------
+             atten_layers: 1,3,5                         
+               batch_size: 1                             
+                    beta1: 0.5                           
+                    beta2: 0.999                         
+          checkpoints_dir: ./checkpoints                 
+           continue_train: False                         
+                crop_size: 256                           
+                 dataroot: /home/openxs/kunyu/datasets/InfraredCity-Lite/Double/Moitor	[default: placeholder]
+             dataset_mode: unaligned_double              	[default: unaligned]
+                direction: AtoB                          
+              display_env: ROMA                          	[default: main]
+             display_freq: 50                            
+               display_id: None                          
+            display_ncols: 4                             
+             display_port: 8097                          
+           display_server: http://localhost              
+          display_winsize: 256                           
+               easy_label: experiment_name               
+                    epoch: latest                        
+              epoch_count: 1                             
+                eta_ratio: 0.1                           
+          evaluation_freq: 5000                          
+        flip_equivariance: False                         
+                 gan_mode: lsgan                         
+                  gpu_ids: 0                             
+                init_gain: 0.02                          
+                init_type: xavier                        
+                 input_nc: 3                             
+                  isTrain: True                          	[default: None]
+               lambda_GAN: 8.0                           	[default: 1.0]
+               lambda_NCE: 8.0                           	[default: 1.0]
+                lambda_SB: 0.1                           
+               lambda_ctn: 1.0                           
+               lambda_inc: 1.0                           
+                   lmda_1: 0.1                           
+                load_size: 286                           
+                       lr: 1e-05                         	[default: 0.0002]
+           lr_decay_iters: 50                            
+                lr_policy: linear                        
+         max_dataset_size: inf                           
+                    model: roma_unsb                     	[default: cut]
+                 n_epochs: 100                           
+           n_epochs_decay: 100                           
+               n_layers_D: 3                             
+                     name: ROMA_UNSB_001                 	[default: experiment_name]
+                    nce_T: 0.07                          
+                  nce_idt: False                         	[default: True]
+nce_includes_all_negatives_from_minibatch: False                         
+               nce_layers: 0,4,8,12,16                   
+                      ndf: 64                            
+                     netD: basic                         
+                     netF: mlp_sample                    
+                  netF_nc: 256                           
+                     netG: resnet_9blocks                
+                      ngf: 64                            
+             no_antialias: False                         
+          no_antialias_up: False                         
+               no_dropout: True                          
+                  no_flip: True                          	[default: False]
+                  no_html: False                         
+                    normD: instance                      
+                    normG: instance                      
+              num_patches: 256                           
+              num_threads: 4                             
+                output_nc: 3                             
+                    phase: train                         
+                pool_size: 0                             
+               preprocess: resize_and_crop               
+          pretrained_name: None                          
+               print_freq: 100                           
+         random_scale_max: 3.0                           
+             save_by_iter: False                         
+          save_epoch_freq: 5                             
+         save_latest_freq: 5000                          
+           serial_batches: False                         
+stylegan2_G_num_downsampling: 1                             
+                   suffix:                               
+         update_html_freq: 1000                          
+                  use_idt: False                         
+                  verbose: False                         
+----------------- End -------------------

models/roma_unsb_model.py

@@ -221,6 +221,7 @@ class RomaUnsbModel(BaseModel):
 
         parser.add_argument('--atten_layers', type=str, default='1,3,5', help='compute Cross-Similarity on which layers')
         
+        parser.add_argument('--tau', type=float, default=0.1, help='used in unsb')
 
         parser.set_defaults(pool_size=0)  # no image pooling