eval batch size added

2026-09-09 11:16:00 +08:00 · 2018-03-09 09:55:15 -08:00
parent cf3188d7da
commit 66bf60800a
2 changed files with 2 additions and 1 deletions
@@ -15,6 +15,7 @@
  "lr": 0.00001875,
  "warmup_steps": 4000,
  "batch_size": 1,
+  "eval_batch_size": 32,
  "r": 5,

  "griffin_lim_iters": 60,
@@ -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)