Skip to content

Commit d7c366c

Browse files
authored
[MagpieTTS] fix val dataloader config: set seed and shard_seed=randomized to ensure deterministic multi-GPU eval (NVIDIA-NeMo#15427)
Signed-off-by: Xuesong Yang <1646669+XuesongYang@users.noreply.github.com>
1 parent a56f63a commit d7c366c

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

examples/tts/conf/magpietts/magpietts_lhotse.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,8 @@ model:
108108
use_bucketing: false
109109
force_finite: true
110110
force_map_dataset: true
111+
seed: 42
112+
shard_seed: "randomized"
111113
drop_last: false
112114
shuffle: false
113115
num_workers: 2

examples/tts/conf/magpietts/magpietts_lhotse_moe.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,8 @@ model:
112112
use_bucketing: false
113113
force_finite: true
114114
force_map_dataset: true
115+
seed: 42
116+
shard_seed: "randomized"
115117
drop_last: false
116118
shuffle: false
117119
num_workers: 2

0 commit comments

Comments
 (0)