↓ 1 callersMethod__init__(
self,
*,
d_model: int ,
num_layer: int,
num_att_head: int,
xmodaler/modeling/decoder/meshed_decoder.py:77
↓ 1 callersMethod__init__(
self,
*,
params,
lr=1e-3,
betas=(0.9, 0.999),
eps=1e-8,
xmodaler/optim/radam.py:12
↓ 1 callersMethod_expand_state(self, states, selected_beam, batch_size, beam_size, cur_beam_size)
xmodaler/modeling/decode_strategy/beam_searcher.py:24
↓ 1 callersMethod_expand_state(self, states, selected_beam, batch_size, beam_size, cur_beam_size)
xmodaler/modeling/decode_strategy/ensemble_beam_searcher.py:26
↓ 1 callersMethod_make_grid dH - offset relative to the ordinate idx, [b, h, w], elem value in [-1, 1] make grid for linear interpolation along H dimension
xmodaler/modeling/layers/tdconved_layers.py:164