| data: | |
| dataroot: /amax/xyf/data/ImageNet/ | |
| img_size: 256 | |
| name: ImageNet | |
| dataloader: | |
| micro_batch: 0 | |
| num_workers: 4 | |
| pin_memory: true | |
| prefetch_factor: 2 | |
| mask: | |
| dir_invert_color: true | |
| dir_path: /amax/xyf/data/NVIDIAIrregularMaskDataset/train/ | |
| mask_type: dir | |
| noise_type: real | |
| real_dataset: | |
| - dataroot: /amax/xyf/data/Places365/ | |
| img_size: 256 | |
| name: Places365 | |
| smooth_kernel_size: 13 | |
| smooth_sigma: 3 | |
| smooth_type: gaussian | |
| model: | |
| mpn: | |
| base_n_channels: 64 | |
| neck_n_channels: 128 | |
| rin: | |
| base_n_channels: 32 | |
| neck_n_channels: 128 | |
| seed: 1234 | |
| train: | |
| batch_size: 4 | |
| coef_adv: 0.001 | |
| coef_idmrf: 0.001 | |
| coef_mpn: 2.0 | |
| coef_rec: 1.4 | |
| coef_semantic: 0.0001 | |
| eval_freq: 1000 | |
| n_steps: 150000 | |
| optim_disc: | |
| betas: | |
| - 0.5 | |
| - 0.9 | |
| lr: 1.0e-05 | |
| optim_mpn: | |
| betas: | |
| - 0.5 | |
| - 0.9 | |
| lr: 1.0e-05 | |
| optim_rin: | |
| betas: | |
| - 0.5 | |
| - 0.9 | |
| lr: 1.0e-05 | |
| pretrained: ./runs/separate_imagenet_maskdir_realnoise/ckpt/step149999/model.pt | |
| print_freq: 100 | |
| resume: null | |
| sample_freq: 1000 | |
| save_freq: 5000 | |