About use_checkpoint
LinShan-Bin opened this issue · comments
The code for use checkpoint misses one parameter:
The original code in network_swinir.py line 399:
x = checkpoint.checkpoint(blk, x)Should be:
x = checkpoint.checkpoint(blk, x, x_size)
Thank you for reporting the bug. We will correct it.