[New] our more recent work on this topic: Towards Robust and Adaptive Motion Forecasting: A Causal Representation Perspective, CVPR 2022. python imitate.py --contrast_weight=2.0 ...