Batch working algorithm

[alexcbb]

Hello, first of all thank you for this very nice work ! I've explored a bit the capabilities with LeGrad and it's quite impressive zero-shot.
I wanted to use the system on top of one of my work to train an external network, but I didn't find out a way to perform the algorithm in a batch (or maybe I misunderstood). Can you help me on this ?

Thank you in advance

[alexcbb]

Hi I've implemented by myself a batch version, it should be working, maybe it needs some verifications.