git clone https://github.com/zhanglejun02/DL-Fall-25-Final-Project.git
pip install -e .pip install -r exp_requirements.txtMore details can be found in torchrun_main.py. Our experiment scripts are tested on Python 3.8 with PyTorch 2.1.
bash run2.shchange the task name to test different tasks