I noticed the GPU memory does not get cleared after successfully running STalign.transform_image_target_to_source, especially with a high resolution image can take up 60GB. Do you know what vectors/matrices are stored on the GPU and not getting removed?