Are you using the Model GPU memory snapshotting for this?