File size: 380 Bytes
9bf4bd7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
_base_ = '_base_abinet-vision.py'

model = dict(
    decoder=dict(
        d_model=512,
        num_iters=3,
        language_decoder=dict(
            type='ABILanguageDecoder',
            d_model=512,
            n_head=8,
            d_inner=2048,
            n_layers=4,
            dropout=0.1,
            detach_tokens=True,
            use_self_attn=False,
        )), )