Skip to content

fix: test dataloader warns batch_size set to 1 but never enforces it - #66

Open
andrewwhitecdw wants to merge 1 commit into
NVlabs:mainfrom
andrewwhitecdw:bugfix/build-a3bb120d
Open

fix: test dataloader warns batch_size set to 1 but never enforces it#66
andrewwhitecdw wants to merge 1 commit into
NVlabs:mainfrom
andrewwhitecdw:bugfix/build-a3bb120d

Conversation

@andrewwhitecdw

Copy link
Copy Markdown

Problem

fix: test dataloader warns batch_size set to 1 but never enforces it

Fix

Replace:

    if batch_size != 1:
        logger.warning(
            "When testing, batch size is set to 1. "
            "This is the only mode that is supported for d2."
        )

    return build_detection_test_loader(

with:

    if batch_size != 1:
        logger.warning(
            "When testing, batch size is set to 1. "
            "This is the only mode that is supported for d2."
        )
        batch_size = 1

    return build_detection_test_loader(

Files changed

  • odise/data/build.py

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