NeurIPs Camera Ready version

Contains the version of the code at the time of publishing the camera ready version.

v20241111

Before we implement the transformer.
Still checking the run_hg_classification.py.