From e29bf123f1c7d9af76cc14734a7dcecf0abccdc5 Mon Sep 17 00:00:00 2001
From: Oscar Pocock <oscar@oscarpo.co>
Date: Wed, 23 Mar 2022 10:41:09 +0000
Subject: [PATCH] Added prediction batch size to config

---
 src/autophotographer/config.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/autophotographer/config.py b/src/autophotographer/config.py
index 52b6857..bfc5c68 100644
--- 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")