Skip to content

[Bugfix] Fix a bug in LLava dataloader#54

Open
PoTsui99 wants to merge 2 commits into
showlab:mainfrom
PoTsui99:main
Open

[Bugfix] Fix a bug in LLava dataloader#54
PoTsui99 wants to merge 2 commits into
showlab:mainfrom
PoTsui99:main

Conversation

@PoTsui99

@PoTsui99 PoTsui99 commented Nov 27, 2024

Copy link
Copy Markdown

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