ABCNet / configs /attn_R_50.yaml
karensanchez's picture
add configuration files
ff044a9
raw
history blame contribute delete
321 Bytes
_BASE_: "Base-TotalText.yaml"
MODEL:
WEIGHTS: "weights/batext/pretrain_attn_R_50.pth"
RESNETS:
DEPTH: 50
BATEXT:
RECOGNIZER: "attn" # "attn" "rnn"
SOLVER:
IMS_PER_BATCH: 8
BASE_LR: 0.001
MAX_ITER: 5000
CHECKPOINT_PERIOD: 1000
TEST:
EVAL_PERIOD: 1000
OUTPUT_DIR: "output/batext/totaltext/attn_R_50"