@AnushreeBannadabhavi We should also try LLTFI on VectorNet, as it is used in Apollo Baidu ADS. Here's the PyTorch implementation of it: https://github.com/Liang-ZX/VectorNet