some tweaks and fixes

2024-05-18 22:27:05 +01:00
parent 9922c1ba85
commit 5d7960d4c7
3 changed files with 13 additions and 9 deletions
--- a/hyperparams.py
+++ b/hyperparams.py
@@ -1,5 +1,5 @@
 FILTER_COUNT = 32
 KERNEL_SIZE = 2
 CROP_SIZE = 224
-BATCH_SIZE = 4
-EPOCHS = 5
+BATCH_SIZE = 256
+EPOCHS = 20