poi_Engineering / src /sts /projects /SWINTS /configs /SWINTS-swin-pretrain.yaml
Juartaurus's picture
Upload folder using huggingface_hub
1865436
raw
history blame contribute delete
477 Bytes
_BASE_: "Base-SWINTS_swin.yaml"
MODEL:
WEIGHTS: "swin_imagenet_pretrain.pth"
SWINTS:
NUM_PROPOSALS: 300
NUM_CLASSES: 2
DATASETS:
# TRAIN: ("totaltext_train",)
TRAIN: ("totaltext_train","icdar_2015_train","icdar_2013_train","icdar_2017_validation_mlt","icdar_2017_mlt","icdar_curvesynthtext_train1","icdar_curvesynthtext_train2",)
TEST: ("totaltext_test",)
SOLVER:
STEPS: (360000,420000)
MAX_ITER: 450000
CHECKPOINT_PERIOD: 10000
INPUT:
FORMAT: "RGB"