tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

Yolov8 想将激活函数修改为relu,但是发现代码写死了,yolov5使用的auto更具有通用性

Open gopin95 opened this issue 2 months ago • 1 comments

Env

  • GPU, e.g. V100, RTX2080, TX2, Xavier NX, Nano, etc.
  • OS, e.g. Ubuntu16.04, Win10, etc.
  • Cuda version
  • TensorRT version

About this repo

  • which branch/tag/commit are you using?
  • which model? yolov5, retinaface?

Your problem

  • what is your command? e.g. sudo ./yolov5 -s
  • what's your output?
  • what output do you expect?

gopin95 avatar Apr 25 '24 02:04 gopin95