Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
nicolai256 authored Oct 24, 2022
1 parent 747688d commit 8a06ede
Showing 1 changed file with 84 additions and 0 deletions.
84 changes: 84 additions & 0 deletions _config/reference_webcam.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,84 @@
# Generator
generator: &generator_j
type: GeneratorJ
args:
use_bias: True
tanh: True
append_smoothers: True
# resnet_blocks: 7
resnet_blocks: 5
# filters: [32, 64, 128, 128, 128, 64]
#filters: [32, 64, 64, 64, 64, 64]
filters: [32, 32, 32, 32, 32, 32]
input_channels: 3


# Optimizer of Generator
opt_generator: &opt_generator
type: Adam
args:
lr: 0.0004
betas: [0.9, 0.999]
weight_decay: 0.00001


# Discriminator
discriminator: &discriminatorn
type: DiscriminatorN_IN
args:
num_filters: 12
n_layers: 2


# Optimizer of Discriminator
opt_discriminator: &opt_discriminator
type: Adam
args:
lr: 0.0004
betas: [0.9, 0.999]
weight_decay: 0.00001


# Parameters of Perception Loss (VGG-Loss)
perception_loss: &perception_loss
weight: 6.0
perception_model:
type: PerceptualVGG19
args:
feature_layers: [0, 3, 5]
use_normalization: False


# Training Parameters
trainer: &trainer_1
batch_size: 16
epochs: 50000000
reconstruction_weight: 4.
adversarial_weight: 0.5
use_image_loss: True
reconstruction_criterion: L1Loss
adversarial_criterion: MSELoss


# Training Dataset Parameters
training_dataset: &training_dataset
type: DatasetPatches_M
dir_pre: input_filtered
dir_post: output
dir_mask: mask
patch_size: 32


# "Main" of this YAML file
job:
training_dataset: *training_dataset
generator: *generator_j
opt_generator: *opt_generator
discriminator: *discriminatorn
opt_discriminator: *opt_discriminator
perception_loss: *perception_loss
trainer: *trainer_1

num_workers: 1
device: "cuda:0"

0 comments on commit 8a06ede

Please sign in to comment.