You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
765 B
41 lines
765 B
---
|
|
experiment_desc: Stripformer_pretrained
|
|
|
|
train:
|
|
files_a: ./datasets/GoPro/train/blur/**/*.png
|
|
files_b: ./datasets/GoPro/train/sharp/**/*.png
|
|
size: &SIZE 256
|
|
crop: random
|
|
preload: &PRELOAD false
|
|
preload_size: &PRELOAD_SIZE 0
|
|
bounds: [0, 1]
|
|
scope: geometric
|
|
|
|
val:
|
|
files_a: ./datasets/GoPro/test/blur/**/*.png
|
|
files_b: ./datasets/GoPro/test/sharp/**/*.png
|
|
size: *SIZE
|
|
scope: None
|
|
crop: random
|
|
preload: *PRELOAD
|
|
preload_size: *PRELOAD_SIZE
|
|
bounds: [0, 1]
|
|
|
|
model:
|
|
g_name: Stripformer
|
|
content_loss: Stripformer_Loss
|
|
|
|
num_epochs: 3000
|
|
train_batches_per_epoch: 2103
|
|
val_batches_per_epoch: 1111
|
|
batch_size: 8
|
|
image_size: [256, 256]
|
|
|
|
optimizer:
|
|
name: adam
|
|
lr: 0.0001
|
|
scheduler:
|
|
name: cosine
|
|
start_epoch: 50
|
|
min_lr: 0.0000001
|