New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

Fix chunksize in utils.py when batch is less than ncpu value #98

Merged

dscripka merged 2 commits into dscripka:main from dlipatov:fix-chunksize

Jan 19, 2024

Contributor

dlipatov commented Jan 14, 2024

Added protection for batch.shape[0]//ncpu to be always greater than zero which happens when training with high number of cpus


          Fix chunksize in utils.py when batch is less than ncpu value

db418ab

dscripka approved these changes

View reviewed changes

Owner

dscripka left a comment

Nice fix, thanks!


          fix flake8 issue

e3f1046

dscripka merged commit dc5a234 into dscripka:main

2 checks passed

# for free to join this conversation on GitHub. Already have an account? # to comment

Labels

None yet