Skip to content
New issue

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

Sign up for GitHub

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

Already on GitHub? Sign in to your account

Jump to bottom

ObjectDetectionTask: explicit batch size #1928

Merged
merged 1 commit into from
Mar 3, 2024
Merged

ObjectDetectionTask: explicit batch size #1928

merged 1 commit into from
Mar 3, 2024

Conversation

adamjstewart
Copy link
Collaborator

Just following the instructions from the error message:

tests/trainers/test_detection.py::TestObjectDetectionTask::test_trainer[True-faster-rcnn-nasa_marine_debris]
  lib/python3.11/site-packages/lightning/pytorch/utilities/data.py:77: Trying to infer the `batch_size` from an ambiguous collection. The batch size we found is 1. To avoid any miscalculations, use `self.log(..., batch_size=batch_size)`.

@adamjstewart adamjstewart requested a review from ashnair1 March 3, 2024 16:10
@github-actions github-actions bot added trainers PyTorch Lightning trainers dependencies Packaging and dependencies labels Mar 3, 2024
@ashnair1 ashnair1 merged commit 2789f29 into microsoft:main Mar 3, 2024
24 checks passed
@adamjstewart adamjstewart deleted the fixes/detection-batch-size branch March 3, 2024 17:00
@adamjstewart adamjstewart modified the milestone: 0.5.3 Mar 4, 2024
@adamjstewart adamjstewart mentioned this pull request Mar 4, 2024
Merged
@adamjstewart adamjstewart added this to the 0.5.3 milestone Mar 4, 2024
@adamjstewart adamjstewart modified the milestones: 0.5.3, 0.6.0 Aug 6, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@ashnair1 ashnair1 ashnair1 approved these changes

Assignees
No one assigned
Labels
dependencies Packaging and dependencies trainers PyTorch Lightning trainers
Projects
None yet
Milestone
0.6.0
Development

Successfully merging this pull request may close these issues.
2 participants
@adamjstewart @ashnair1