Strided Checkpoint Doesn't work as expected in 2d #51
Description
The current way the stride is calculated for the checkpoint is broken in 2d (but works for 3d)
{{ message }}
The current way the stride is calculated for the checkpoint is broken in 2d (but works for 3d)