class

GS::MASt3R::MASt3RDecoder

Inherits Reference < Object

Full MASt3R Decoder Has two sets of blocks for processing image pairs

Constructors

new(config : MASt3RDecoderConfig, device : Tensor::Device = Tensor::Device::GPU)
Source

Instance methods

blocks
Source
blocks2
Source
config
Source
embed_proj
Source
forward_pair(enc1 : Autograd::Variable, enc2 : Autograd::Variable) : Tuple(Autograd::Variable, Autograd::Variable)

Decode pair of encoder outputs with cross-attention enc1, enc2: [batch, seq, encoder_dim] Returns: {dec1, dec2} each [batch, seq, embed_dim]

Source
forward_single(encoder_out : Autograd::Variable) : Autograd::Variable

Decode single encoder output encoder_out: [batch, seq, encoder_dim] Returns: [batch, seq, embed_dim]

Source
mask_token
Source
norm
Source
parameters
Source