memory burst

[hczyni]

Dear developer, during my training process, video memory gradually climbed until the video memory burst. How to solve this problem? I really want to read your article, but I can't find it online

[linkenfaqiu]

Hello! I apologize for the delayed reply. The issue you mentioned with the video memory burst may occur when the GPU's available memory is exceeded by the network's memory requirements. Recently, I updated my code to reduce the network's channels to a quarter of their original size.

Original network's detail: Params: 35,652,628, MACs: 229,696,335,660, Memory: 31,868 MiB.
After reduction's detail: Params: 3,193,092, MACs: 20,737,401,756, Memory: 7,156 MiB.

You can use my updated code, and I truly appreciate you taking the time to read through my project. Thank you!