almost same as mmedit
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user