search

erfanzar / EasyDeL

Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
https://easydel.readthedocs.io/en/latest/
Apache License 2.0
167 stars 19 forks source link

Fix eval batch loop #49

Closed w11wo closed 7 months ago

w11wo commented 7 months ago

Fixes evaluation loop, otherwise raises ValueError: Incompatible shapes for broadcasting due to shapes mismatch.

erfanzar commented 7 months ago

thanks for your PR can you make this change and open PR in beta branch instead of Main ?