gitextract_98xwsw9y/ ├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── configs/ │ ├── train/ │ │ └── train_scalable.yaml │ └── validation/ │ └── validation_scalable.yaml ├── data_preprocess.py ├── environment.yml ├── pyproject.toml ├── requirements.txt ├── scripts/ │ ├── install_pyg.sh │ └── traj_clstering.py ├── smart/ │ ├── __init__.py │ ├── datamodules/ │ │ ├── __init__.py │ │ └── scalable_datamodule.py │ ├── datasets/ │ │ ├── __init__.py │ │ ├── preprocess.py │ │ └── scalable_dataset.py │ ├── layers/ │ │ ├── __init__.py │ │ ├── attention_layer.py │ │ ├── fourier_embedding.py │ │ └── mlp_layer.py │ ├── metrics/ │ │ ├── __init__.py │ │ ├── average_meter.py │ │ ├── min_ade.py │ │ ├── min_fde.py │ │ ├── next_token_cls.py │ │ └── utils.py │ ├── model/ │ │ ├── __init__.py │ │ └── smart.py │ ├── modules/ │ │ ├── __init__.py │ │ ├── agent_decoder.py │ │ ├── map_decoder.py │ │ └── smart_decoder.py │ ├── preprocess/ │ │ ├── __init__.py │ │ └── preprocess.py │ ├── tokens/ │ │ ├── __init__.py │ │ ├── cluster_frame_5_2048.pkl │ │ └── map_traj_token5.pkl │ ├── transforms/ │ │ ├── __init__.py │ │ └── target_builder.py │ └── utils/ │ ├── __init__.py │ ├── cluster_reader.py │ ├── config.py │ ├── geometry.py │ ├── graph.py │ ├── list.py │ ├── log.py │ ├── nan_checker.py │ └── weight_init.py ├── train.py └── val.py