Trajectron-plus-plus/trajectron/model/components/__init__.py
2020-04-05 21:43:49 -04: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