BUG: _check_input gives the wrong information if the batch dimensions are wrong

lhackel-tub / ConfigILM

A Library for configurable combination of pre-configured and possibly pre-trained Image and Language Models

https://lhackel-tub.github.io/ConfigILM/

MIT License

32 stars 4 forks source link

Open lhackel-tub opened 1 week ago

lhackel-tub commented 1 week ago

Instead of printing the length of the batch shape (len(batch.shape)) it prints the length of the batch (len(batch))