feat: add code for finetuning moondream

2024-12-08 17:33:37 +00:00
parent 4bf2c89bf8
commit 6b53cb0411
11 changed files with 261 additions and 0 deletions
--- a/resnet/hyperparams.py
+++ b/resnet/hyperparams.py
@@ -0,0 +1,5 @@
+FILTER_COUNT = 32
+KERNEL_SIZE = 2
+CROP_SIZE = 224
+BATCH_SIZE = 256
+EPOCHS = 20