2025-03-07 01:42:12,033 INFO: 
                ____                _       _____  ____
               / __ ) ____ _ _____ (_)_____/ ___/ / __ \
              / __  |/ __ `// ___// // ___/\__ \ / /_/ /
             / /_/ // /_/ /(__  )/ // /__ ___/ // _, _/
            /_____/ \__,_//____//_/ \___//____//_/ |_|
     ______                   __   __                 __      __
    / ____/____   ____   ____/ /  / /   __  __ _____ / /__   / /
   / / __ / __ \ / __ \ / __  /  / /   / / / // ___// //_/  / /
  / /_/ // /_/ // /_/ // /_/ /  / /___/ /_/ // /__ / /<    /_/
  \____/ \____/ \____/ \____/  /_____/\____/ \___//_/|_|  (_)
    
Version Information: 
	BasicSR: 1.2.0+cc63400
	PyTorch: 2.4.1
	TorchVision: 0.19.1+cu121
2025-03-07 01:42:12,033 INFO: 
  name: RetinexFormer_LOL_v1
  model_type: ImageCleanModel
  scale: 1
  num_gpu: 2
  manual_seed: 100
  datasets:[
    train:[
      name: TrainSet
      type: Dataset_PairedImage
      dataroot_gt: data/LOLv1/Train/target
      dataroot_lq: data/LOLv1/Train/input
      geometric_augs: True
      filename_tmpl: {}
      io_backend:[
        type: disk
      ]
      use_shuffle: True
      num_worker_per_gpu: 8
      batch_size_per_gpu: 128
      mini_batch_sizes: [128]
      iters: [300000]
      gt_size: 128
      gt_sizes: [128]
      dataset_enlarge_ratio: 1
      prefetch_mode: None
      phase: train
      scale: 1
    ]
    val:[
      name: ValSet
      type: Dataset_PairedImage
      dataroot_gt: data/LOLv1/Test/target
      dataroot_lq: data/LOLv1/Test/input
      io_backend:[
        type: disk
      ]
      phase: val
      scale: 1
    ]
  ]
  network_g:[
    type: RetinexFormer
    in_channels: 3
    out_channels: 3
    n_feat: 40
    stage: 1
    num_blocks: [1, 2, 2]
  ]
  path:[
    pretrain_network_g: /home/tiennv/chaos/repo/Retinexformer/pretrain/LOL_v1.pth
    strict_load_g: True
    resume_state: None
    root: /mnt/Userdrive/tiennv/chaos/repo/Retinexformer
    experiments_root: /mnt/Userdrive/tiennv/chaos/repo/Retinexformer/experiments/RetinexFormer_LOL_v1
    models: /mnt/Userdrive/tiennv/chaos/repo/Retinexformer/experiments/RetinexFormer_LOL_v1/models
    training_states: /mnt/Userdrive/tiennv/chaos/repo/Retinexformer/experiments/RetinexFormer_LOL_v1/training_states
    log: /mnt/Userdrive/tiennv/chaos/repo/Retinexformer/experiments/RetinexFormer_LOL_v1
    visualization: /mnt/Userdrive/tiennv/chaos/repo/Retinexformer/experiments/RetinexFormer_LOL_v1/visualization
  ]
  train:[
    total_iter: 150000
    warmup_iter: -1
    use_grad_clip: True
    scheduler:[
      type: CosineAnnealingRestartCyclicLR
      periods: [46000, 104000]
      restart_weights: [1, 1]
      eta_mins: [0.0003, 1e-06]
    ]
    mixing_augs:[
      mixup: True
      mixup_beta: 1.2
      use_identity: True
    ]
    optim_g:[
      type: Adam
      lr: 0.0002
      betas: [0.9, 0.999]
    ]
    pixel_opt:[
      type: L1Loss
      loss_weight: 1
      reduction: mean
    ]
  ]
  val:[
    window_size: 4
    val_freq: 1000.0
    save_img: False
    rgb2bgr: True
    use_image: False
    max_minibatch: 8
    metrics:[
      psnr:[
        type: calculate_psnr
        crop_border: 0
        test_y_channel: False
      ]
    ]
  ]
  logger:[
    print_freq: 500
    save_checkpoint_freq: 1000.0
    use_tb_logger: True
    wandb:[
      project: low_light
      resume_id: None
    ]
  ]
  dist_params:[
    backend: nccl
    port: 29500
  ]
  is_train: True
  dist: True
  rank: 0
  world_size: 2
  rename_flag: False

2025-03-07 01:42:12,103 INFO: Dataset Dataset_PairedImage - TrainSet is created.
2025-03-07 01:42:12,103 INFO: Training statistics:
	Number of train images: 3380
	Dataset enlarge ratio: 1
	Batch size per gpu: 128
	World size (gpu number): 2
	Require iter number per epoch: 14
	Total epochs: 10715; iters: 150000.
2025-03-07 01:42:12,131 INFO: Dataset Dataset_PairedImage - ValSet is created.
2025-03-07 01:42:12,131 INFO: Number of val images/folders in ValSet: 3380
2025-03-07 01:42:12,373 INFO: Loading RetinexFormer model from /home/tiennv/chaos/repo/Retinexformer/pretrain/LOL_v1.pth.
2025-03-07 01:42:12,392 INFO: Model [ImageCleanModel] is created.