update modules and pipeline

This commit is contained in:
hofee
2024-08-21 17:57:52 +08:00
parent 913d4e521d
commit 837e1c870a
8 changed files with 129 additions and 57 deletions

View File

@@ -0,0 +1,12 @@
from abc import abstractmethod
from torch import nn
class SequenceEncoder(nn.Module):
def __init__(self):
super(SequenceEncoder, self).__init__()
@abstractmethod
def encode_sequence(self, pts_embedding_list, pose_embedding_list):
pass

View File

@@ -0,0 +1,10 @@
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