implement ResNet model

2024-05-18 01:07:06 +01:00
parent eaa6cf38ca
commit 0ca7d5b8ec
3 changed files with 194 additions and 4 deletions
--- a/hyperparams.py
+++ b/hyperparams.py
@@ -1,5 +1,5 @@
 FILTER_COUNT = 32
 KERNEL_SIZE = 2
-CROP_SIZE = 200
+CROP_SIZE = 224
 BATCH_SIZE = 4
-EPOCHS = 20
+EPOCHS = 5