singan_interp-pad_balloons.py 254 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
8
9
10
_base_ = ['../singan/singan_balloons.py']

model = dict(
    type='PESinGAN',
    generator=dict(
        type='SinGANMSGeneratorPE', interp_pad=True, noise_with_pad=True))

train_cfg = dict(fixed_noise_with_pad=True)

dist_params = dict(backend='nccl')