104 lines
1.8 KiB
YAML
104 lines
1.8 KiB
YAML
name: horse2zebra
|
|
engine: cyclegan
|
|
result_dir: ./result
|
|
max_iteration: 18000
|
|
|
|
distributed:
|
|
model:
|
|
# broadcast_buffers: False
|
|
|
|
misc:
|
|
random_seed: 1004
|
|
|
|
checkpoints:
|
|
interval: 2000
|
|
|
|
log:
|
|
logger:
|
|
level: 20 # DEBUG(10) INFO(20)
|
|
|
|
model:
|
|
generator:
|
|
_type: ResGenerator
|
|
in_channels: 3
|
|
out_channels: 3
|
|
base_channels: 64
|
|
num_blocks: 9
|
|
padding_mode: reflect
|
|
norm_type: IN
|
|
use_dropout: False
|
|
discriminator:
|
|
_type: PatchDiscriminator
|
|
_distributed:
|
|
bn_to_syncbn: True
|
|
in_channels: 3
|
|
base_channels: 64
|
|
num_conv: 3
|
|
norm_type: BN
|
|
|
|
loss:
|
|
gan:
|
|
loss_type: lsgan
|
|
weight: 1.0
|
|
real_label_val: 1.0
|
|
fake_label_val: 0.0
|
|
cycle:
|
|
level: 1
|
|
weight: 10.0
|
|
id:
|
|
level: 1
|
|
weight: 0
|
|
|
|
optimizers:
|
|
generator:
|
|
_type: Adam
|
|
lr: 2e-4
|
|
betas: [0.5, 0.999]
|
|
discriminator:
|
|
_type: Adam
|
|
lr: 2e-4
|
|
betas: [0.5, 0.999]
|
|
|
|
data:
|
|
train:
|
|
buffer_size: 50
|
|
dataloader:
|
|
batch_size: 16
|
|
shuffle: True
|
|
num_workers: 4
|
|
pin_memory: True
|
|
drop_last: True
|
|
dataset:
|
|
_type: GenerationUnpairedDataset
|
|
root_a: "/data/i2i/horse2zebra/trainA"
|
|
root_b: "/data/i2i/horse2zebra/trainB"
|
|
random_pair: True
|
|
pipeline:
|
|
- Load
|
|
- Resize:
|
|
size: [286, 286]
|
|
- RandomCrop:
|
|
size: [256, 256]
|
|
- RandomHorizontalFlip
|
|
- ToTensor
|
|
scheduler:
|
|
start: 9000
|
|
target_lr: 0
|
|
test:
|
|
dataloader:
|
|
batch_size: 4
|
|
shuffle: False
|
|
num_workers: 1
|
|
pin_memory: False
|
|
drop_last: False
|
|
dataset:
|
|
_type: GenerationUnpairedDataset
|
|
root_a: "/data/i2i/horse2zebra/testA"
|
|
root_b: "/data/i2i/horse2zebra/testB"
|
|
random_pair: False
|
|
pipeline:
|
|
- Load
|
|
- Resize:
|
|
size: [256, 256]
|
|
- ToTensor
|