> For the complete documentation index, see [llms.txt](https://docs.dobb-e.com/llms.txt). Markdown versions of documentation pages are available by appending `.md` to page URLs; this page is available as [Markdown](https://docs.dobb-e.com/software/readme-1/optional-training-your-own-home-pretrained-representations.md).

# \[Optional] Training Your Own Home Pretrained Representations

This step assumes you have already downloaded the HoNY RGB + actions dataset to somewhere on your machine, and have updated your `configs/env_vars/env_vars.yaml` accordingly.

* **Single GPU:** To reproduce our HPR encoder, run in terminal:

  ```bash
  python train.py --config-name=train_moco
  ```
* Multi-GPU: We use huggingface 🤗 accelerate to run multi-GPU training. Run the following commands

  ```bash
  accelerate config # Interactively walk you through setting up multi-GPU training
  accelerate launch train.py --config-name=train_moco
  ```
