Method with_pos_embed

ADHMR/lib/models/scorenet.py:94–95

(self, tensor, pos)

        self.dropout3 = nn.Dropout(p=self.dropout_rate)
        self.activation = nn.ReLU()
    def with_pos_embed(self, tensor, pos):
        return tensor + pos
    
    def forward(self, tgt, memory,mask= None,mask_ctx = None,pos= None,pos_ctx=None):
        tgt2 = self.norm1(tgt)