/yolov5-on-sagemaker

YoloV5 on SageMaker, including bring your own container

Primary LanguageJupyter Notebook

yolov5-on-sagemaker

YoloV5 on SageMaker, including bring your own container

We use the AWS deeplearning pytorch container to include the yolov5 and let it can train on SageMaker environment.

The available deeplearning container images are listing here