search

xuebinqin / U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
Apache License 2.0
8.31k stars 1.43k forks source link

对u2netp模型进行qat量化 #362

Open ZHIZIHUABU opened 11 months ago

ZHIZIHUABU commented 11 months ago

使用pytorch的量化工具fx对u2netp进行qat量化，首先加载了训练好的fp32模型，然后按照流程对u2netp模型进行qat，loss随着epoch迭代越来越大，miou越来越小，qat后的模型完全错误。