iSEE-Laboratory / DiffUIR

The official implementation of the paper of CVPR2024: Selective Hourglass Mapping for Universal Image Restoration Based on Diffusion Model
MIT License
105 stars 7 forks source link

Params和FLOPS #18

Closed pb-lang closed 4 weeks ago

pb-lang commented 1 month ago

作者您好,请问您的代码怎么计算Params和FLOPS

zhengdian1 commented 1 month ago

我手头也没有存这部分的代码,但是就是正常的测试流程,FLOPS就按3步的计算开销算