delete all abstract class

This commit is contained in:
hofee
2024-08-23 13:04:38 +08:00
parent eceedd5c15
commit 2503fca572
8 changed files with 4 additions and 44 deletions

20
modules/pose_encoder.py Normal file
View File

@@ -0,0 +1,20 @@
from torch import nn
import PytorchBoot.stereotype as stereotype
@stereotype.module("pose_encoder")
class PoseEncoder(nn.Module):
def __init__(self, config):
super(PoseEncoder, self).__init__()
self.config = config
pose_dim = config["pose_dim"]
self.act = nn.ReLU(True)
self.pose_encoder = nn.Sequential(
nn.Linear(pose_dim, 256),
self.act,
nn.Linear(256, 256),
self.act,
)
def encode_pose(self, pose):
return self.pose_encoder(pose)