Team Ai
Apppublic

eihab2342/code-efficiency

sourceHugging Faceupdated 5mo agoView on Hugging Face
0likes
config.py27 linesDownload Raw Back to utils
1# src/utils/config.py2from pathlib import Path3 4BASE_DIR = Path(__file__).resolve().parent.parent.parent   # 5 6# ── Data ──────────────────────────────────────────────7DATA_RAW_DIR       = BASE_DIR / "data" / "raw"8DATA_PROCESSED_DIR = BASE_DIR / "data" / "processed"9 10# ── Model ─────────────────────────────────────────────11MODEL_DIR          = BASE_DIR / "models"12PRETRAINED_MODEL   = "Salesforce/codet5-small"     13FINETUNED_MODEL    = MODEL_DIR / "codet5-finetuned"14 15MAX_INPUT_LEN   = 51216MAX_TARGET_LEN  = 51217BATCH_SIZE      = 818EPOCHS          = 319LEARNING_RATE   = 5e-520TRAIN_SPLIT     = 0.821VAL_SPLIT       = 0.122TEST_SPLIT      = 0.123 24# ── API ───────────────────────────────────────────────25API_HOST = "0.0.0.0"26API_PORT = 800027