Docker Image

How to Start

To make it more convenient for you to use the HAT algorithm package, we have provided a command to start the Docker environment with just one click, which greatly simplifies the process of environment setup.

After you obtain the release package, first navigate to the path where the release package is located. Then, follow the instructions in the Environment Deployment section to complete the installation of the basic Docker environment, add the user to the Docker group, and pull the Docker image.

After completing the above operations, you only need to use the following command to set up all the environments required for successfully running the HAT samples with just one click.

docker run -it --shm-size="15g" -v `pwd`:/open_explorer openexplorer/ai_toolchain_ubuntu_22_j6_gpu:{version}
Note

The version number {version} here is only an example, please replace it with the real version number of the image you have obtained.

where openexplorer/ai_toolchain_ubuntu_22_j6_gpu:{version} is the image name and -v is used to mount the local path to the docker path.

Get Execution Script

After successfully starting the docker environment, get the compressed file from the release package and extract the configs and tools folders.

Once you have all the environments, you can follow the training tutorial to train a fixed-point model using HAT step by step.