Fixing QA generation

lboegner · Jul 19, 2023 · 4a6e14a · 4a6e14a
1 parent 6ae0f8f
commit 4a6e14a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchsig/datasets/conf.py b/torchsig/datasets/conf.py
@@ -124,7 +124,7 @@ class WidebandSig53CleanValConfig(WidebandSig53CleanTrainConfig):
 
 @dataclass
 class WidebandSig53CleanValQAConfig(WidebandSig53CleanValConfig):
-    num_samples: int = 25
+    num_samples: int = 250
 
 
 @dataclass
@@ -137,7 +137,7 @@ class WidebandSig53ImpairedTrainConfig(WidebandSig53Config):
 
 @dataclass
 class WidebandSig53ImpairedTrainQAConfig(WidebandSig53ImpairedTrainConfig):
-    num_samples: int = 2_50
+    num_samples: int = 250
 
 
 @dataclass
@@ -149,4 +149,4 @@ class WidebandSig53ImpairedValConfig(WidebandSig53ImpairedTrainConfig):
 
 @dataclass
 class WidebandSig53ImpairedValQAConfig(WidebandSig53ImpairedValConfig):
-    num_samples: int = 25
+    num_samples: int = 250