from torch import nn import PytorchBoot.stereotype as stereotype @stereotype.module("transformer_seq_encoder") class TransformerSequenceEncoder(nn.Module): def __init__(self, config): super(TransformerSequenceEncoder, self).__init__() self.config = config def encode_sequence(self, pts_embedding_list, pose_embedding_list): pass