Image Segmentation
English
ISDNet-pytorch / config.py
Antoine1091's picture
Upload folder using huggingface_hub
a585f5a verified
raw
history blame contribute delete
415 Bytes
"""
ISDNet Configuration
"""
# Data paths
DATA_ROOT = "/ccast/FLAIR1024_optimal"
STDC_PRETRAIN_PATH = "STDCNet813M_73.91.tar"
# Training hyperparameters
BATCH_SIZE_PER_GPU = 16
NUM_WORKERS = 4
BASE_LR = 1e-3
WEIGHT_DECAY = 0.0005
NUM_EPOCHS = 80
# Model configuration
NUM_CLASSES = 15 # Classes 0-14 only
CROP_SIZE = 512
DOWN_RATIO = 4
IGNORE_INDEX = 255 # For classes >= 15
# Checkpointing
SAVE_INTERVAL = 5