Skip to content

Conversation

@PoTsui99
Copy link

@PoTsui99 PoTsui99 commented Nov 27, 2024

Leaving drop_last as the default value of False may cause an issue where the batch size is incorrectly assumed to be batch_size_mmu when calculating loss_mmu. For comparison, the text2image dataloader drop the last incomlete batch.

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.

1 participant