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

feat: ✨ PyTorch Recognition Model Multi-GPU support #1164

Merged
merged 10 commits into from
Mar 31, 2023
Merged

feat: ✨ PyTorch Recognition Model Multi-GPU support #1164

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
12ce5a3
feat: :sparkles: PyTorch Recognition Model Multi-GPU support
odulcy-mindee Mar 9, 2023
f496592
Update
odulcy-mindee Mar 9, 2023
422d6a6
Update
odulcy-mindee Mar 9, 2023
2dcb6e1
Fix wrong device object
odulcy-mindee Mar 9, 2023
bef86a2
Progress bar only on `rank == 0`
odulcy-mindee Mar 9, 2023
131a168
Use DistributedSampler
odulcy-mindee Mar 28, 2023
8052489
Clean imports
odulcy-mindee Mar 28, 2023
11c3fa7
lint: :lipstick:
odulcy-mindee Mar 30, 2023
5198e68
Mypy
odulcy-mindee Mar 31, 2023
86a1dd0
isort
odulcy-mindee Mar 31, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 1 addition & 7 deletions doctr/models/factory/hub.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,13 +13,7 @@
from pathlib import Path
from typing import Any

from huggingface_hub import ( # type: ignore[attr-defined]
HfApi,
HfFolder,
Repository,
hf_hub_download,
snapshot_download,
)
from huggingface_hub import HfApi, HfFolder, Repository, hf_hub_download, snapshot_download

from doctr import models
from doctr.file_utils import is_tf_available, is_torch_available
Expand Down
Loading