Spaces:
Running
Running
add frcrn model
Browse files
examples/frcrn/yaml/config.yaml
CHANGED
|
@@ -4,6 +4,11 @@ model_name: "frcrn"
|
|
| 4 |
num_gpus: -1
|
| 5 |
|
| 6 |
lr: 0.001
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
max_epochs: 100
|
| 8 |
weight_decay: 1.0e-05
|
| 9 |
clip_grad_norm: 10.0
|
|
|
|
| 4 |
num_gpus: -1
|
| 5 |
|
| 6 |
lr: 0.001
|
| 7 |
+
lr_scheduler: "CosineAnnealingLR"
|
| 8 |
+
lr_scheduler_kwargs:
|
| 9 |
+
T_max: 250000
|
| 10 |
+
eta_min: 0.0001
|
| 11 |
+
|
| 12 |
max_epochs: 100
|
| 13 |
weight_decay: 1.0e-05
|
| 14 |
clip_grad_norm: 10.0
|