> 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

Completely optional step, if you want to tinker with the pretrained representation or train your own model.

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
  ```