Added prediction batch size to config

2022-03-23 10:41:09 +00:00 · 2022-03-23 10:41:09 +00:00 · e29bf123f1
commit e29bf123f1
parent 68ff4ea0ad
1 changed files with 1 additions and 0 deletions
--- a/src/autophotographer/config.py
+++ b/src/autophotographer/config.py
@ -14,6 +14,7 @@ PRED_BATCH_SIZE = 4
 EPOCHS = 20
 LR = 0.001
 IMAGE_SIZE = 32
+PRED_BATCH_SIZE = 4

 WARMUP_PLOT = os.path.join("output", "plot.png")
 WARMUP_MODEL = os.path.join("output", "plot.pth")