NVIDIA-AI-IOT/yolov5_gpu_optimization

Yolov5 change silu to relu accuracy problem after exporting to onnx.

maro-jeon opened this issue · 3 comments

Hi,
I saw your study about changing activation function from silu to relu.
The score shows that ReLU mAP is 55.7 (swish pretrained).
so I just changed swish activation to relu activation without any fine-tuning process.

from
image
to
image

Then, the result of accuracy was zero.

Can you describe more about how to achieve the score 55.7 (swish pretrained) for reproducing the score.

Thanks,
Maro JEON

You have to retrain the model.

Thanks, you just fine tune ? or train many epochs ?

Train for 300 epochs.