Skip to content

Commit 38539e3

Browse files
authored
Update train_nsf_sim_cache_sid_load_pretrain.py
1 parent 27c7e3f commit 38539e3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train_nsf_sim_cache_sid_load_pretrain.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def run(rank, n_gpus, hps):
6868
else:train_dataset = TextAudioLoader(hps.data.training_files, hps.data)
6969
train_sampler = DistributedBucketSampler(
7070
train_dataset,
71-
hps.train.batch_size,
71+
hps.train.batch_size*n_gpus,
7272
# [100, 200, 300, 400, 500, 600, 700, 800, 900, 1000, 1200,1400], # 16s
7373
[100, 200, 300, 400, 500, 600, 700, 800, 900], # 16s
7474
num_replicas=n_gpus,

0 commit comments

Comments
 (0)