HoneyTian commited on
Commit
f176576
·
1 Parent(s): 4ce1c90

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