almost same as mmedit

This commit is contained in:
2020-08-08 13:17:26 +08:00
parent 7cf235781d
commit a5133e6795
3 changed files with 61 additions and 18 deletions

View File

@@ -1,14 +1,14 @@
name: horse2zebra
engine: cyclegan
result_dir: ./result
max_iteration: 18000
max_iteration: 16600
distributed:
model:
# broadcast_buffers: False
misc:
random_seed: 1004
random_seed: 324
checkpoints:
interval: 2000
@@ -29,12 +29,12 @@ model:
use_dropout: False
discriminator:
_type: PatchDiscriminator
_distributed:
bn_to_syncbn: True
# _distributed:
# bn_to_syncbn: False
in_channels: 3
base_channels: 64
num_conv: 3
norm_type: BN
norm_type: IN
loss:
gan:
@@ -82,7 +82,7 @@ data:
- RandomHorizontalFlip
- ToTensor
scheduler:
start: 9000
start: 8300
target_lr: 0
test:
dataloader: