Trajectron-plus-plus/data/nuScenes/models/baseline/model_registrar-1.pt