We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 27c7e3f commit 38539e3Copy full SHA for 38539e3
1 file changed
train_nsf_sim_cache_sid_load_pretrain.py
@@ -68,7 +68,7 @@ def run(rank, n_gpus, hps):
68
else:train_dataset = TextAudioLoader(hps.data.training_files, hps.data)
69
train_sampler = DistributedBucketSampler(
70
train_dataset,
71
- hps.train.batch_size,
+ hps.train.batch_size*n_gpus,
72
# [100, 200, 300, 400, 500, 600, 700, 800, 900, 1000, 1200,1400], # 16s
73
[100, 200, 300, 400, 500, 600, 700, 800, 900], # 16s
74
num_replicas=n_gpus,
0 commit comments