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

5 lines
194 B
Python
Raw Permalink Normal View History

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