24GB graphics card memory is not enough

Hello, I tried to use 3090 to run your network, and found that the video memory was insufficient. The picture size is the same. Is there any way to reduce the video memory usage? I see that your server is 4090, which is also 24GB. If you have time, please reply

Please make sure that the "use_checkpoint" is set as "True".
Turning on the checkpoint function will reduce more than 50% GPU memory consumption.

Please make sure that the "use_checkpoint" is set as "True". Turning on the checkpoint function will reduce more than 50% GPU memory consumption.

Thank you very much for your answer. I keep the parameter setting of the network, 'use_checkpoint' defaults to True. Is there any other setting

Registration-CorrMLP/CorrMLP/networks.py

Line 22 in da5ce37

use_checkpoint: bool = True):

This might be because using the checkpointing function in 4090 can reduce more GPU memory than 3090.
Perhaps you can reduce the 'enc_channels' and 'dec_channels' a little bit.

Ok, I'll try. Thank you