-
Notifications
You must be signed in to change notification settings - Fork 3
/
default.yaml
132 lines (132 loc) · 2.89 KB
/
default.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
seed_everything: 123
trainer:
logger: null
checkpoint_callback: null
enable_checkpointing: true
callbacks:
- class_path: pytorch_lightning.callbacks.RichModelSummary
init_args:
max_depth: -1
- class_path: pytorch_lightning.callbacks.RichProgressBar
init_args:
refresh_rate: 1
leave: false
theme:
description: white
progress_bar: '#6206E0'
progress_bar_finished: '#6206E0'
progress_bar_pulse: '#6206E0'
batch_progress: white
time: grey54
processing_speed: grey70
metrics: white
console_kwargs: null
- class_path: pytorch_lightning.callbacks.ModelCheckpoint
init_args:
dirpath: null
filename: null
monitor: val/loss
verbose: false
save_last: true
save_top_k: 1
save_weights_only: false
mode: min
auto_insert_metric_name: true
every_n_train_steps: null
train_time_interval: null
every_n_epochs: null
save_on_train_epoch_end: null
default_root_dir: null
gradient_clip_val: null
gradient_clip_algorithm: null
process_position: 0
num_nodes: 1
num_processes: null
devices: auto
gpus: null
auto_select_gpus: false
tpu_cores: null
ipus: null
log_gpu_memory: null
progress_bar_refresh_rate: null
enable_progress_bar: true
overfit_batches: 0.0
track_grad_norm: -1
check_val_every_n_epoch: 1
fast_dev_run: false
accumulate_grad_batches: null
max_epochs: 50
min_epochs: 10
max_steps: -1
min_steps: null
max_time: null
limit_train_batches: null
limit_val_batches: null
limit_test_batches: null
limit_predict_batches: null
val_check_interval: null
flush_logs_every_n_steps: null
log_every_n_steps: 50
accelerator: gpu
strategy: ddp_find_unused_parameters_false
sync_batchnorm: true
precision: 32
enable_model_summary: true
weights_summary: top
weights_save_path: null
num_sanity_val_steps: 0
resume_from_checkpoint: null
profiler: null
benchmark: null
deterministic: null
reload_dataloaders_every_n_epochs: 0
auto_lr_find: false
replace_sampler_ddp: true
detect_anomaly: false
auto_scale_batch_size: false
prepare_data_per_node: null
plugins: null
amp_backend: native
amp_level: null
move_metrics_to_cpu: false
multiple_trainloader_mode: max_size_cycle
stochastic_weight_avg: false
terminate_on_nan: null
data:
class_path: src.datamodules.sr_datamodule.SRDataModule
init_args:
root: ./data/
trainsets:
- - DIV2K
- train
trainsets_repeat: 20
testsets:
- - DIV2K
- train
- - benchmark
- B100
- - benchmark
- Set5
- - benchmark
- Set14
- - benchmark
- Urban100
batch_size: 4
train_scales:
- 2
- 3
- 4
test_scales:
- 2
- 2.5
- 3
- 3.5
- 4
- 6
- 8
- 10
- 15
- 20
patch_size: 48
num_workers: 16
pin_memory: False