Trajectron-plus-plus/trajectron/model/components/__init__.py

5 lines
194 B
Python
Raw Permalink Normal View History

2020-01-13 18:55:45 +00:00
from .discrete_latent import DiscreteLatent
from .gmm2d import GMM2D
from .map_encoder import CNNMapEncoder
from .additive_attention import AdditiveAttention, TemporallyBatchedAdditiveAttention