Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
nvidia
/
mit-b5
like
6
Image Classification
Transformers
PyTorch
google-tensorflow
TensorFlow
imagenet_1k
segformer
vision
Inference Endpoints
arxiv:
2105.15203
License:
other
Model card
Files
Files and versions
Community
2
Train
Deploy
Use this model
refs/pr/1
mit-b5
2 contributors
History:
8 commits
amyeroberts
HF staff
Add TF weights
4dc413c
about 2 years ago
.gitattributes
1.18 kB
initial commit
almost 3 years ago
README.md
3.24 kB
Update README.md
over 2 years ago
config.json
70 kB
Remove reshape_last_stage
over 2 years ago
preprocessor_config.json
272 Bytes
Update preprocessor_config.json
almost 3 years ago
pytorch_model.bin
pickle
Detected Pickle imports (3)
"torch._utils._rebuild_tensor_v2"
,
"torch.FloatStorage"
,
"collections.OrderedDict"
What is a pickle import?
328 MB
LFS
First commit
almost 3 years ago
tf_model.h5
329 MB
LFS
Add TF weights
about 2 years ago