stevenwjy:

Do you perhaps have any thoughts on what might cause such a different in memory usage?


No, since I’m seeing the same memory usage and thus cannot reproduce the issue.
Could you post your output of both runs for the first ~10 steps?