Spaces:
Running
Running
add frcrn model
Browse files
toolbox/torchaudio/models/frcrn/configuration_frcrn.py
CHANGED
@@ -13,6 +13,9 @@ class FRCRNConfig(PretrainedConfig):
|
|
13 |
num_gpus: int = -1,
|
14 |
|
15 |
lr: float = 0.001,
|
|
|
|
|
|
|
16 |
max_epochs: int = 100,
|
17 |
weight_decay: float = 0.00001,
|
18 |
clip_grad_norm: float = 10.,
|
@@ -40,6 +43,9 @@ class FRCRNConfig(PretrainedConfig):
|
|
40 |
self.num_gpus = num_gpus
|
41 |
|
42 |
self.lr = lr
|
|
|
|
|
|
|
43 |
self.max_epochs = max_epochs
|
44 |
self.weight_decay = weight_decay
|
45 |
self.clip_grad_norm = clip_grad_norm
|
|
|
13 |
num_gpus: int = -1,
|
14 |
|
15 |
lr: float = 0.001,
|
16 |
+
lr_scheduler: str = "CosineAnnealingLR",
|
17 |
+
lr_scheduler_kwargs: dict = None,
|
18 |
+
|
19 |
max_epochs: int = 100,
|
20 |
weight_decay: float = 0.00001,
|
21 |
clip_grad_norm: float = 10.,
|
|
|
43 |
self.num_gpus = num_gpus
|
44 |
|
45 |
self.lr = lr
|
46 |
+
self.lr_scheduler = lr_scheduler
|
47 |
+
self.lr_scheduler_kwargs = lr_scheduler_kwargs or dict()
|
48 |
+
|
49 |
self.max_epochs = max_epochs
|
50 |
self.weight_decay = weight_decay
|
51 |
self.clip_grad_norm = clip_grad_norm
|