Update train_nsf_sim_cache_sid_load_pretrain.py

2023-04-09 15:26:05 +00:00 · 2023-04-09 15:26:05 +00:00 · 38539e3c07
parent 27c7e3f6a3
commit 38539e3c07
1 changed files with 1 additions and 1 deletions
--- a/train_nsf_sim_cache_sid_load_pretrain.py
+++ b/train_nsf_sim_cache_sid_load_pretrain.py
@ -68,7 +68,7 @@ def run(rank, n_gpus, hps):
    else:train_dataset = TextAudioLoader(hps.data.training_files, hps.data)
    train_sampler = DistributedBucketSampler(
        train_dataset,
-        hps.train.batch_size,
+        hps.train.batch_size*n_gpus,
        # [100, 200, 300, 400, 500, 600, 700, 800, 900, 1000, 1200,1400],  # 16s
        [100, 200, 300, 400, 500, 600, 700, 800, 900],  # 16s
        num_replicas=n_gpus,