eschmidbauer/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity
c8ab89d243
DL-Art-School/codes/options/train/train_vix_resgenv2.yml

94 lines
1.9 KiB
YAML
Raw Normal View History

Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
#### general settings
Config updates
2020-05-13 21:27:49 +00:00
name: train_vix_resgenv2
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
use_tb_logger: true
model: srgan
distortion: sr
scale: 4
gpu_ids: [0]
amp_opt_level: O1
#### datasets
datasets:
train:
name: vixcloseup
mode: LQGT
Config file update
2020-05-12 16:07:58 +00:00
dataroot_GT: E:\4k6k\datasets\vixen\4k_closeup\hr
Config updates
2020-05-13 21:27:49 +00:00
dataroot_LQ: E:\4k6k\datasets\vixen\4k_closeup\lr_corrupted
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
doCrop: false
use_shuffle: true
Config updates
2020-05-13 21:27:49 +00:00
n_workers: 0 # per GPU
batch_size: 8
Config file update
2020-05-12 16:07:58 +00:00
target_size: 192
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
color: RGB
val:
name: adrianna_val
mode: LQGT
dataroot_GT: E:\4k6k\datasets\adrianna\val\hhq
dataroot_LQ: E:\4k6k\datasets\adrianna\val\hr
#### network structures
network_G:
Config updates
2020-05-13 21:27:49 +00:00
which_model_G: ResGenV2
nf: 256
nb_denoiser: 20
nb_upsampler: 10
upscale_applications: 0
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
network_D:
which_model_D: discriminator_resnet_passthrough
Config updates
2020-05-13 21:27:49 +00:00
nf: 64
# LR corruption network.
network_C:
which_model_G: ResGenV2
nf: 192
nb_denoiser: 20
nb_upsampler: 0
upscale_applications: 0
inject_noise: False
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
#### path
path:
Random config changes
2020-05-06 23:25:48 +00:00
#pretrain_model_G: ../experiments/pretrained_resnet_G.pth
Fix megabatch scaling, log low and med-res gen images
2020-05-05 14:34:57 +00:00
#pretrain_model_D: ~
Config updates
2020-05-13 21:27:49 +00:00
pretrain_model_C: ../experiments/pretrained_corruptors/resgen_xl_noise_19000.pth
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
strict_load: true
Config updates
2020-05-13 21:27:49 +00:00
resume_state: ~
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
#### training settings: learning rate scheme, loss
train:
Config updates
2020-05-13 21:27:49 +00:00
lr_G: !!float 1e-4
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
weight_decay_G: 0
beta1_G: 0.9
beta2_G: 0.99
Config updates
2020-05-13 21:27:49 +00:00
lr_D: !!float 1e-4
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
weight_decay_D: 0
beta1_D: 0.9
beta2_D: 0.99
lr_scheme: MultiStepLR
niter: 400000
warmup_iter: -1 # no warm up
Config file update
2020-05-12 16:07:58 +00:00
lr_steps: [20000, 60000, 80000, 100000]
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
lr_gamma: 0.5
Config file update
2020-05-12 16:07:58 +00:00
mega_batch_factor: 2
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
pixel_criterion: l1
Config updates
2020-05-13 21:27:49 +00:00
pixel_weight: .01
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
feature_criterion: l1
feature_weight: 1
feature_weight_decay: 1
feature_weight_decay_steps: 500
feature_weight_minimum: 1
gan_type: gan # gan | ragan
Config updates
2020-05-13 21:27:49 +00:00
gan_weight: .01
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
Config file update
2020-05-12 16:07:58 +00:00
D_update_ratio: 1
Random config changes
2020-05-06 23:25:48 +00:00
D_init_iters: 0
Add support for passthrough disc/gen Add RRDBNetXL, which performs processing at multiple image sizes. Add DiscResnet_passthrough, which allows passthrough of image at different sizes for discrimination. Adjust the rest of the repo to allow generators that return more than just a single image.
2020-05-04 20:01:43 +00:00
manual_seed: 10
val_freq: !!float 5e2
#### logger
logger:
print_freq: 50
save_checkpoint_freq: !!float 5e2
Reference in New Issue Copy Permalink