Trajectron-plus-plus/code/model/components/__init__.py
2020-01-13 10:55:45 -08:00

4 lines
194 B
Python

from .discrete_latent import DiscreteLatent
from .gmm2d import GMM2D
from .map_encoder import CNNMapEncoder
from .additive_attention import AdditiveAttention, TemporallyBatchedAdditiveAttention