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

The validation set cannot be larger than the batch size at the moment, but this shouldn't be necessary

Describe the solution you'd like

Allow the validation set(s) to be larger than the batch size

Describe alternatives you've considered

No response

Additional context

No response