11 lines
362 B
Python
11 lines
362 B
Python
|
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
|