eval batch size added

2018-03-09 09:55:15 -08:00 · 2018-03-09 09:55:15 -08:00 · 332abb1de6
parent a65712d163
commit 332abb1de6
2 changed files with 2 additions and 1 deletions
--- a/config.json
+++ b/config.json
@ -15,6 +15,7 @@
  "lr": 0.00001875,
  "warmup_steps": 4000,
  "batch_size": 1,
+  "eval_batch_size": 32,
  "r": 5,

  "griffin_lim_iters": 60,
--- a/train.py
+++ b/train.py
@ -326,7 +326,7 @@ def main(args):
                              c.power
                             )

-    val_loader = DataLoader(val_dataset, batch_size=c.batch_size,
+    val_loader = DataLoader(val_dataset, batch_size=c.eval_batch_size,
                            shuffle=False, collate_fn=val_dataset.collate_fn,
                            drop_last=False, num_workers= 4,
                            pin_memory=True)