Skip to content

Latest commit

 

History

History
30 lines (23 loc) · 996 Bytes

README.md

File metadata and controls

30 lines (23 loc) · 996 Bytes

Object Detection Example

This folder contains an example script which demonstrates the usage of DETR to run object detection task on Gaudi platform.

Single-HPU inference

PT_HPU_LAZY_MODE=1 python3 run_example.py \
	--model_name_or_path facebook/detr-resnet-101 \
	--image_path "http://images.cocodataset.org/val2017/000000039769.jpg" \
	--use_hpu_graphs \
	--bf16 \
	--print_result