Ah ok. Thank you Simon for your help. I really do appreciate it! If I wanted to figure out exactly why the batch size messes up the backward pass, how would I debug this? I can’t just use print statements on the forward pass right? Would it involve hooks?