JacobLinCool commited on
Commit
771996d
1 Parent(s): 38548f2

perf: set batch size to 32

Browse files
Files changed (2) hide show
  1. config.json +3 -3
  2. infer/modules/train/train.py +1 -1
config.json CHANGED
@@ -58,7 +58,7 @@
58
  "use_spectral_norm": false
59
  },
60
  "train": {
61
- "batch_size": 4,
62
  "betas": [
63
  0.8,
64
  0.99
@@ -67,7 +67,7 @@
67
  "c_mel": 45,
68
  "epochs": 20000,
69
  "eps": 1e-09,
70
- "fp16_run": true,
71
  "init_lr_ratio": 1,
72
  "learning_rate": 0.0001,
73
  "log_interval": 200,
@@ -76,4 +76,4 @@
76
  "segment_size": 12800,
77
  "warmup_epochs": 0
78
  }
79
- }
 
58
  "use_spectral_norm": false
59
  },
60
  "train": {
61
+ "batch_size": 32,
62
  "betas": [
63
  0.8,
64
  0.99
 
67
  "c_mel": 45,
68
  "epochs": 20000,
69
  "eps": 1e-09,
70
+ "fp16_run": false,
71
  "init_lr_ratio": 1,
72
  "learning_rate": 0.0001,
73
  "log_interval": 200,
 
76
  "segment_size": 12800,
77
  "warmup_epochs": 0
78
  }
79
+ }
infer/modules/train/train.py CHANGED
@@ -98,7 +98,7 @@ def train(exp_dir: str):
98
  hps.pretrainD = "assets/pretrained_v2/f0D40k.pth"
99
  hps.version = "v2"
100
  hps.gpus = "0"
101
- hps.train.batch_size = 8
102
  hps.sample_rate = "40k"
103
  hps.if_f0 = 1
104
  hps.if_latest = 0
 
98
  hps.pretrainD = "assets/pretrained_v2/f0D40k.pth"
99
  hps.version = "v2"
100
  hps.gpus = "0"
101
+ hps.train.batch_size = 32
102
  hps.sample_rate = "40k"
103
  hps.if_f0 = 1
104
  hps.if_latest = 0