Hey did you solve this issue ever? I think i might be having a similar issue where if i run my model on just a batch of