MuhammedKsee commited on
Commit
a11ccb2
·
verified ·
1 Parent(s): 62e8bb2

Upload config.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.py +19 -0
config.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+
3
+ class HParams:
4
+ # Model Mimari Ayarları (Model ağırlıkları ile eşleşmesi için AYNEN kalmalı)
5
+ IMAGE_SIZE = 224
6
+ MAX_TOKENS = 77
7
+ VISION_WIDTH = 768
8
+ TEXT_WIDTH = 512
9
+ EMBED_DIM = 512
10
+ VISION_LAYERS = 12
11
+ TEXT_LAYERS = 12
12
+ VISION_HEADS = 12
13
+ TEXT_HEADS = 8
14
+ FFN_MULT = 4
15
+
16
+ # Test Ayarları
17
+ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
18
+ MODEL_PATH = "best_model.pt" # Hugging Face'den indirilen model
19
+ TEST_IMAGE = "images/test_image.jpg" # Test edilecek örnek resim