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
521 stars 39 forks source link

can't compute flops using torchsummaryX #72

Closed luwang03 closed 6 months ago

luwang03 commented 6 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 6 months ago

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

luwang03 commented 6 months ago

I changed nothing in the denoising network.

Algolzw commented 6 months ago

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

luwang03 commented 6 months ago

Actually, my torchsummaryX is1.3.0.

luwang03 commented 6 months ago

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

luwang03 commented 6 months ago

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

Algolzw commented 6 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 6 months ago

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

Algolzw commented 6 months ago

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

luwang03 commented 6 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.