Spaces:
Running
Running
| arch: | |
| _target_: arch.gcir.gcir_nano | |
| pretrained: false | |
| version: 1 | |
| callbacks: | |
| model_checkpoint: | |
| _target_: callbacks.default.CustomModelCheckpoint | |
| filename: '{epoch:03d}-{step:07d}' | |
| every_n_train_steps: 1000 | |
| save_top_k: -1 | |
| save_last: true | |
| dirpath: checkpoints | |
| auto_insert_metric_name: false | |
| verbose: true | |
| lmodule: | |
| _target_: lmodule.sr_lmodule.SRLightningModule | |
| hparams: | |
| lpips_net: alex | |
| l1_weight: 1 | |
| p_weight: 1 | |
| lr: ${lr} | |
| betas: ${betas} | |
| eps: ${eps} | |
| weight_decay: ${weight_decay} | |
| milestones: ${milestones} | |
| gamma: ${gamma} | |
| datamodule: | |
| _target_: datamodule.sr_datamodule.SRDataModule | |
| train_dataset: | |
| _target_: dataset.sr_dataset.BlindSRDataset | |
| _convert_: partial | |
| hq_data_dir: ${train_hq_data_dir} | |
| sr_scale: ${sr_scale} | |
| n_channels: 3 | |
| degradation_type: ${degradation_type} | |
| shuffle_prob: 0.1 | |
| use_sharp: false | |
| hq_patch_size: 256 | |
| lq_patch_size: 64 | |
| val_dataset: | |
| _target_: dataset.sr_dataset.BlindSRDataset | |
| _convert_: partial | |
| hq_data_dir: ${val_data_dir} | |
| sr_scale: ${sr_scale} | |
| n_channels: 3 | |
| degradation_type: ${degradation_type} | |
| shuffle_prob: 0.1 | |
| use_sharp: false | |
| hq_patch_size: 256 | |
| lq_patch_size: 64 | |
| batch_size: ${batch_size} | |
| num_workers: ${num_workers} | |
| num_val_workers: 8 | |
| iterations_per_epoch: 1000 | |
| use_random_sampler: false | |
| trainer: | |
| _target_: pytorch_lightning.Trainer | |
| accelerator: gpu | |
| strategy: | |
| _target_: pytorch_lightning.plugins.training_type.ddp.DDPPlugin | |
| find_unused_parameters: true | |
| gpus: ${gpus} | |
| precision: 32 | |
| max_steps: 100000 | |
| check_val_every_n_epoch: 10 | |
| replace_sampler_ddp: false | |
| benchmark: true | |
| model: gcir_base | |
| name: gcir_base | |
| version: v1_SRscale2 | |
| sr_scale: 2 | |
| gpus: | |
| - 0 | |
| - 1 | |
| train_hq_data_dir: /home/cll/datasets/swinir_train | |
| val_data_dir: /home/cll/datasets/swinir_test | |
| epochs: 300 | |
| warmup_epochs: 20 | |
| cooldown_epochs: 10 | |
| batch_size: 8 | |
| num_workers: 8 | |
| optimizer_name: adamw | |
| lr: 0.0001 | |
| betas: | |
| - 0.9 | |
| - 0.999 | |
| eps: 1.0e-08 | |
| weight_decay: 0.05 | |
| milestones: | |
| - 50000 | |
| - 100000 | |
| - 150000 | |
| - 200000 | |
| - 300000 | |
| gamma: 0.5 | |
| degradation_type: bsrgan | |
| checkpoint_path: null | |
| use_channels_last: false | |