search

Algolzw / image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.
https://algolzw.github.io/ir-sde/index.html
MIT License
580 stars 42 forks source link

can't compute flops using torchsummaryX #72

Closed luwang03 closed 11 months ago

luwang03 commented 11 months ago

When I computed the flops of model, the code as:

image

, I met an error as: image

How can I solve this problem?

Algolzw commented 11 months ago

The code seems correct and I didn't meet this problem. Did you change something in the denoising network?

luwang03 commented 11 months ago

I changed nothing in the denoising network.

Algolzw commented 11 months ago

Maybe it's the version problem. Can you upgrade your torchsummaryX to 1.3.0?

luwang03 commented 11 months ago

Actually, my torchsummaryX is1.3.0.

luwang03 commented 11 months ago

For thehe error in image, I printed the 'a' is image

luwang03 commented 11 months ago

It seems the 'a' array has '-' entry.

Algolzw commented 11 months ago

= =.

It's really weird because the code works very well for me. Screenshot 2023-12-19 at 10 43 06

Maybe it doesn't work on Windows?

luwang03 commented 11 months ago

Maybe you are right! Windows can't work it.

Algolzw commented 11 months ago

Yes. Or can you try to add the __init__ function to SimpleGate and add super().__init__() to it as well?

luwang03 commented 11 months ago

Sorry for my late reply. I tried init function, Windows also can't work. Maybe I need to run the codes on linux. Thanks for your help.