# Training


Logic for training a ParaGuide model. See ``train.sh`` for a training template.

In our work, we fine-tune the original SSD-LM diffusion model (https://huggingface.co/xhan77/ssdlm), with our modified approach/architecture.

``train.py`` contains our training code.
