| # yaml-language-server: $schema=https://raw.githubusercontent.com/the-database/traiNNer-redux/refs/heads/master/schemas/redux-config.schema.json | |
| #################### | |
| # General Settings | |
| #################### | |
| name: %scale%_%archname% | |
| scale: %scale% | |
| use_amp: true # Speed up training and reduce VRAM usage. NVIDIA only. | |
| amp_bf16: false # Use bf16 instead of fp16 for AMP, RTX 3000 series or newer only. | |
| num_gpu: auto | |
| use_channels_last: true | |
| ################################# | |
| # Dataset and Dataloader Settings | |
| ################################# | |
| datasets: | |
| val: | |
| name: Val Dataset | |
| type: singleimagedataset | |
| # Path of the LR (low res) images to run the model on. | |
| dataroot_lq: [ | |
| datasets/val/dataset1/lr, | |
| datasets/val/dataset1/lr2 | |
| ] | |
| io_backend: | |
| type: disk | |
| ##################### | |
| # Network Settings | |
| ##################### | |
| # Generator model settings | |
| network_g: | |
| type: %archname% | |
| ############################ | |
| # Pretrain and Resume Paths | |
| ############################ | |
| path: | |
| # Path of the model to run on the validation images. | |
| pretrain_network_g: experiments/%scale%_%archname%/models/net_g_ema_1000.safetensors | |
| ############# | |
| # Validation | |
| ############# | |
| val: | |
| val_enabled: true # Whether to enable validations. If disabled, all validation settings below are ignored. | |
| save_img: true # Whether to save the validation images during validation, in the results/<name>/visualization folder. | |
| tile_size: 0 # Tile size of input, reduce VRAM usage but slower inference. 0 to disable. | |
| tile_overlap: 8 # Number of pixels to overlap tiles by, larger is slower but reduces tile seams. | |
| suffix: ~ # Add suffix to saved images, if None, use exp name. | |
| metrics_enabled: false # Whether to run metrics calculations during validation. | |
| metrics: | |
| psnr: | |
| type: calculate_psnr | |
| crop_border: 4 | |
| test_y_channel: true | |
| ssim: | |
| type: calculate_ssim | |
| crop_border: 4 # Whether to crop border during validation. | |
| test_y_channel: true # Whether to convert to Y(CbCr) for validation. | |
| topiq: | |
| type: calculate_topiq | |
| lpips: | |
| type: calculate_lpips | |
| better: lower | |
| dists: | |
| type: calculate_dists | |
| better: lower | |