NTIRE 2025 Efficient Burst HDR and Restoration Forum

Go back to competition Back to thread list Post in this thread

> GPU memory overflow

FlopCountAnalysis causes GPU memory overflow when testing the model on a V100. Will submitting tests resolve this issue?

Posted by: m-wq @ March 22, 2025, 8:20 a.m.

Let us test your code and model on our resource. If there are any errors, we will contact you further.

Posted by: Sangmin-Lee @ March 22, 2025, 8:30 a.m.

For this, we have two solutions:
1. use patch inference (256*256) with calculated FLOPs*18 to get the final FLOPs
2. use CPU mode (very long time)
Does the organizer approve of this?

Posted by: THQiu @ March 22, 2025, 8:31 a.m.

We think the first suggestion looks reasonable.
But if possible, we try to test your model for whole image on our resources to check the FLOPs.

Posted by: Sangmin-Lee @ March 22, 2025, 8:34 a.m.

Could you please share the VRAM capacity of the test resources? I'm quite curious. Thank you!

Posted by: m-wq @ March 22, 2025, 8:39 a.m.

We have A100 with capacity 80G.

Posted by: Sangmin-Lee @ March 22, 2025, 9:39 a.m.

Alright, I have no further questions. Thank you very much!

Posted by: m-wq @ March 22, 2025, 9:42 a.m.
Post in this thread