config.py

import torch
from pathlib import Path

# Definisikan ROOT path proyek (folder batik_vision_project)
ROOT_PATH = Path(__file__).resolve().parent.parent 

# Path ke data
DATA_PATH = ROOT_PATH / "Batik-Indonesia" # <-- GANTI BARIS INI

# Hyperparameters
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
BATCH_SIZE = 32  # Dikurangi untuk laptop
IMAGE_SIZE = 224 # Ukuran input untuk ViT/Swin
LEARNING_RATE = 1e-4
EPOCHS = 50  # Dikurangi untuk testing awal

# Pengaturan split
TEST_SPLIT_SIZE = 0.2 # 20% untuk validasi
RANDOM_SEED = 42 # Agar hasil split selalu sama

# Daftar model yang akan diuji
# Mulai dengan model terkecil dulu untuk testing
MODEL_LIST = ["convnext_tiny"]  # Model terkecil untuk testing awal