| 2025-03-07 01:45:31,011 INFO: |
| ____ _ _____ ____ |
| / __ ) ____ _ _____ (_)_____/ ___/ / __ \ |
| / __ |/ __ `// ___// // ___/\__ \ / /_/ / |
| / /_/ // /_/ /(__ )/ // /__ ___/ // _, _/ |
| /_____/ \__,_//____//_/ \___//____//_/ |_| |
| ______ __ __ __ __ |
| / ____/____ ____ ____/ / / / __ __ _____ / /__ / / |
| / / __ / __ \ / __ \ / __ / / / / / / // ___// //_/ / / |
| / /_/ // /_/ // /_/ // /_/ / / /___/ /_/ // /__ / /< /_/ |
| \____/ \____/ \____/ \____/ /_____/\____/ \___//_/|_| (_) |
| |
| Version Information: |
| BasicSR: 1.2.0+cc63400 |
| PyTorch: 2.4.1 |
| TorchVision: 0.19.1+cu121 |
| 2025-03-07 01:45:31,011 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: False |
| rank: 0 |
| world_size: 1 |
| rename_flag: False |
|
|
| 2025-03-07 01:45:31,076 INFO: Dataset Dataset_PairedImage - TrainSet is created. |
| 2025-03-07 01:45:31,076 INFO: Training statistics: |
| Number of train images: 3380 |
| Dataset enlarge ratio: 1 |
| Batch size per gpu: 128 |
| World size (gpu number): 1 |
| Require iter number per epoch: 27 |
| Total epochs: 5556; iters: 150000. |
| 2025-03-07 01:45:31,104 INFO: Dataset Dataset_PairedImage - ValSet is created. |
| 2025-03-07 01:45:31,104 INFO: Number of val images/folders in ValSet: 3380 |
| 2025-03-07 01:45:31,228 INFO: Loading RetinexFormer model from /home/tiennv/chaos/repo/Retinexformer/pretrain/LOL_v1.pth. |
| 2025-03-07 01:45:31,247 INFO: Model [ImageCleanModel] is created. |
| 2025-03-07 01:45:31,442 INFO: Start training from epoch: 0, iter: 0 |
|
|
