File size: 603 Bytes
fe64bad |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from src.models.architectures.transformer import Encoder_TRANSFORMER, Decoder_TRANSFORMER
from src.models.modeltype.motionclip import MOTIONCLIP
JOINTSTYPES = ["a2m", "a2mpl", "smpl", "vibe", "vertices"]
LOSSES = ["rc", "rcxyz", "vel", "velxyz"] # not used: "hp", "mmd", "vel", "velxyz"
def get_model(parameters, clip_model):
encoder = Encoder_TRANSFORMER(**parameters)
decoder = Decoder_TRANSFORMER(**parameters)
parameters["outputxyz"] = "rcxyz" in parameters["lambdas"]
return MOTIONCLIP(encoder, decoder, clip_model=clip_model, **parameters).to(parameters["device"])
|