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

Add DETR #11653

Merged
merged 31 commits into from
Jun 9, 2021
Merged

Add DETR #11653

Changes from 1 commit
Commits
Show all changes
31 commits
Select commit Hold shift + click to select a range
d21c87b
Squash all commits of modeling_detr_v7 branch into one
NielsRogge May 10, 2021
8e06e0b
Improve docs
NielsRogge May 10, 2021
f2592fc
Fix tests
LysandreJik May 10, 2021
1c76900
Style
LysandreJik May 10, 2021
645801d
Improve docs some more and fix most tests
NielsRogge May 10, 2021
0a02d46
Fix slow tests of ViT, DeiT and DETR
NielsRogge May 10, 2021
82507da
Improve replacement of batch norm
NielsRogge May 14, 2021
eb4ee5c
Restructure timm backbone forward
NielsRogge May 14, 2021
9f23454
Make DetrForSegmentation support any timm backbone
NielsRogge May 14, 2021
cfc4b4b
Fix name of output
NielsRogge May 14, 2021
7ddb65a
Address most comments by @LysandreJik
NielsRogge May 21, 2021
a4dec31
Give better names for variables
NielsRogge May 21, 2021
844a280
Conditional imports + timm in setup.py
LysandreJik May 27, 2021
0c5d9ea
Address additional comments by @sgugger
NielsRogge May 28, 2021
f9d379d
Make style, add require_timm and require_vision to testsé
NielsRogge May 28, 2021
0d3eb3c
Remove train_backbone attribute of DetrConfig, add methods to freeze/…
NielsRogge May 28, 2021
9acf820
Add png files to fixtures
NielsRogge May 28, 2021
5d05e45
Fix type hint
NielsRogge May 28, 2021
db9c555
Add timm to workflows
LysandreJik Jun 1, 2021
7bef619
Add `BatchNorm2d` to the weight initialization
LysandreJik Jun 1, 2021
681d714
Fix retain_grad test
NielsRogge Jun 1, 2021
c5ef475
Replace model checkpoints by Facebook namespace
NielsRogge Jun 1, 2021
fdc7edb
Fix name of checkpoint in test
NielsRogge Jun 1, 2021
457b26d
Add user-friendly message when scipy is not available
NielsRogge Jun 1, 2021
734adf9
Address most comments by @patrickvonplaten
NielsRogge Jun 2, 2021
5237b13
Remove return_intermediate_layers attribute of DetrConfig and simplif…
NielsRogge Jun 3, 2021
674297d
Better initialization
LysandreJik Jun 7, 2021
fb30f72
Scipy is necessary to get sklearn metrics
LysandreJik Jun 7, 2021
731297d
Rename TimmBackbone to DetrTimmConvEncoder and rename DetrJoiner to D…
NielsRogge Jun 8, 2021
dcbd458
Make style
NielsRogge Jun 8, 2021
2479544
Improve docs and add 2 community notebooks
NielsRogge Jun 9, 2021
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
Prev Previous commit
Next Next commit
Scipy is necessary to get sklearn metrics
  • Loading branch information
LysandreJik authored and NielsRogge committed Jun 8, 2021
commit fb30f725cda88f0547ec2649ab41cb82eee57977
2 changes: 1 addition & 1 deletion src/transformers/file_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -332,7 +332,7 @@ def is_scipy_available():
def is_sklearn_available():
if importlib.util.find_spec("sklearn") is None:
return False
return importlib.util.find_spec("sklearn.metrics") and is_scipy_available()
return is_scipy_available() and importlib.util.find_spec("sklearn.metrics")


def is_sentencepiece_available():
Expand Down