Change batch_size to batchsize_per_gpu

[clessig]

Is your feature request related to a problem? Please describe.

batch_size is confusing with ddp

Describe the solution you'd like

Replace batchsize with batchsize_per_gpu

Describe alternatives you've considered

No response

Additional context

No response

Organisation

No response