본문 바로가기

Deep Learning

Deformable DETR attention operation cuda build

반응형
  • CUDA operation
cd ./models/ops
sh ./make.sh
# unit test (should see all checking is True)
python test.py
  • Requirement
pip install -r requirements.txt

 

  • Point Pillars
cd ops
python setup.py develop