Parameters on which the model was trained
DATASET:
TRAIN_PATH: "./cityscapes_dataset/train"
TEST_PATH: "./cityscapes_dataset/test"
IMAGE_FORMAT: ".png"
SAM:
CHECKPOINT: "./sam_vit_h_4b8939.pth"
RANK: 512
TRAIN:
BATCH_SIZE: 1
NUM_EPOCHS: 10
LEARNING_RATE: 0.0001
PATIENCE: 10