Remove duplicates, smaller bs by default

This commit is contained in:
enhuiz 2023-01-12 19:50:12 +08:00
parent 777f9d6a13
commit 4c1695e2cf

View File

@ -13,10 +13,6 @@ class Config(ConfigBase):
data_dirs: list[Path] = field(default_factory=lambda: [])
test_data_dirs: list[Path] = field(default_factory=lambda: [])
batch_size: int = 24
eval_batch_size: int = 12
nj: int = 8
@property
def sample_rate(self):
return 24_000
@ -26,6 +22,7 @@ class Config(ConfigBase):
token_dim: int = 256
num_tokens: int = 1024
nj: int = 8
batch_size: int = 128
eval_batch_size: int = 512
warmup_min_lr: float = 1e-6