Spaces:
Running
Running
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():
|