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.

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

Speed up detection tests #1148

Merged
merged 1 commit into from
Feb 26, 2023
Merged

Speed up detection tests #1148

merged 1 commit into from
Feb 26, 2023

Conversation

calebrob6
Copy link
Member

Same idea as #344, but for the object detection tests.

See #1088 for discussion.

On my machine this reduced the time taken for tests/trainers/test_detection.py from 22 seconds to 11 seconds.

@github-actions github-actions bot added testing Continuous integration testing trainers PyTorch Lightning trainers labels Feb 26, 2023
@calebrob6 calebrob6 added this to the 0.4.1 milestone Feb 26, 2023
@adamjstewart
Copy link
Collaborator

This should also save on memory. The difference on Linux isn't huge, but it's much bigger on macOS/Windows.

@adamjstewart adamjstewart merged commit 2db44a4 into main Feb 26, 2023
@adamjstewart adamjstewart deleted the tests/detection branch February 26, 2023 21:36
calebrob6 added a commit that referenced this pull request Apr 10, 2023
yichiac pushed a commit to yichiac/torchgeo that referenced this pull request Apr 29, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
testing Continuous integration testing trainers PyTorch Lightning trainers
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants