HoneyTian commited on
Commit
c079021
·
1 Parent(s): 75f7547

add frcrn model

Browse files
examples/frcrn/yaml/config.yaml CHANGED
@@ -30,3 +30,4 @@ max_snr_db: 20
30
 
31
  num_workers: 4
32
  batch_size: 4
 
 
30
 
31
  num_workers: 4
32
  batch_size: 4
33
+ eval_steps: 25000
toolbox/torchaudio/models/frcrn/configuration_frcrn.py CHANGED
@@ -37,6 +37,8 @@ class FRCRNConfig(PretrainedConfig):
37
 
38
  num_workers: int = 4,
39
  batch_size: int = 4,
 
 
40
  **kwargs
41
  ):
42
  super(FRCRNConfig, self).__init__(**kwargs)
@@ -67,6 +69,7 @@ class FRCRNConfig(PretrainedConfig):
67
 
68
  self.num_workers = num_workers
69
  self.batch_size = batch_size
 
70
 
71
 
72
  def main():
 
37
 
38
  num_workers: int = 4,
39
  batch_size: int = 4,
40
+ eval_steps: int = 25000,
41
+
42
  **kwargs
43
  ):
44
  super(FRCRNConfig, self).__init__(**kwargs)
 
69
 
70
  self.num_workers = num_workers
71
  self.batch_size = batch_size
72
+ self.eval_steps = eval_steps
73
 
74
 
75
  def main():