diff --git a/cfg/extraction.cfg b/cfg/extraction.cfg
index e18d5c0d..66cb15f8 100644
--- a/cfg/extraction.cfg
+++ b/cfg/extraction.cfg
@@ -1,6 +1,12 @@
 [net]
-batch=128
+# Training
+# batch=128
+# subdivisions=4
+
+# Testing
+batch=1
 subdivisions=1
+
 height=224
 width=224
 max_crop=320