You mean why it prints:
Batch size 256: …
not Batch size 2048: …
That's likely because the 2048 triggered the "except" due to exceeding the memory here.
It only prints the successful ones.

→ View original post on X — @rasbt,