unicl-img-recog-demo / configs /unicl_swin_base.yaml
jwyang
first commit
0b36c03
raw
history blame
312 Bytes
MODEL:
NAME: unicl_swin_base
DIM_PROJECTION: 512
IMAGE_ENCODER:
TYPE: swin
DROP_PATH_RATE: 0.5
SWIN:
EMBED_DIM: 128
DEPTHS: [ 2, 2, 18, 2 ]
NUM_HEADS: [ 4, 8, 16, 32 ]
WINDOW_SIZE: 7
TEXT_ENCODER:
NAME: 'transformer'
WIDTH: 512
HEADS: 8
LAYERS: 12