Spaces:
Sleeping
Sleeping
File size: 547 Bytes
9bf4bd7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
_base_ = ['satrn_shallow_5e_st_mj.py']
model = dict(
backbone=dict(type='ShallowCNN', input_channels=3, hidden_dim=256),
encoder=dict(
type='SATRNEncoder',
n_layers=6,
n_head=8,
d_k=256 // 8,
d_v=256 // 8,
d_model=256,
n_position=100,
d_inner=256 * 4,
dropout=0.1),
decoder=dict(
type='NRTRDecoder',
n_layers=6,
d_embedding=256,
n_head=8,
d_model=256,
d_inner=256 * 4,
d_k=256 // 8,
d_v=256 // 8))
|