Skip to content

Commit

Permalink
abcnetv2: Totaltext finetune
Browse files Browse the repository at this point in the history
  • Loading branch information
Yuliang-Liu authored Sep 12, 2021
1 parent d4234ea commit 1ed3107
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions configs/BAText/TotalText/v2_attn_R_50.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
_BASE_: "Base-TotalText.yaml"
MODEL:
WEIGHTS: "https://dl.fbaipublicfiles.com/detectron/ImageNetPretrained/MSRA/R-50.pkl"
BACKBONE:
NAME: "build_fcos_resnet_bifpn_backbone"
BiFPN:
IN_FEATURES: ["res2", "res3", "res4", "res5"]
OUT_CHANNELS: 256
NUM_REPEATS: 2
NORM: "SyncBN"
RESNETS:
DEPTH: 50
BATEXT:
RECOGNIZER: "attn"
USE_COORDCONV: True
USE_AET: True
SOLVER:
IMS_PER_BATCH: 8
BASE_LR: 0.001
MAX_ITER: 5000
CHECKPOINT_PERIOD: 1000
TEST:
EVAL_PERIOD: 1000
OUTPUT_DIR: "output/batext/pretrain/v2_attn_R_50"

0 comments on commit 1ed3107

Please sign in to comment.