Skip to content

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

Merged
dscripka merged 2 commits intodscripka:mainfrom
dlipatov:fix-chunksize
Jan 19, 2024
Merged

Fix chunksize in utils.py when batch is less than ncpu value#98
dscripka merged 2 commits intodscripka:mainfrom
dlipatov:fix-chunksize

Conversation

@dlipatov
Copy link
Copy Markdown
Contributor

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

Copy link
Copy Markdown
Owner

@dscripka dscripka left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice fix, thanks!

@dscripka dscripka merged commit dc5a234 into dscripka:main Jan 19, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants