.. |
_pruned
|
…
|
|
layers
|
Update README, prep for version 0.9.0 release
|
2023-05-11 15:22:50 -07:00 |
__init__.py
|
Add RepViT models
|
2023-07-21 14:56:53 +08:00 |
_builder.py
|
Throw when pretrained weights not available and pretrained=True (principle of least surprise).
|
2023-05-10 10:44:34 -07:00 |
_efficientnet_blocks.py
|
Clean more feature extract issues
|
2023-06-14 14:46:22 -07:00 |
_efficientnet_builder.py
|
Clean more feature extract issues
|
2023-06-14 14:46:22 -07:00 |
_factory.py
|
Merge pull request #1643 from nateraw/docstrings-update
|
2023-05-09 21:33:20 -07:00 |
_features.py
|
Clean more feature extract issues
|
2023-06-14 14:46:22 -07:00 |
_features_fx.py
|
Clean more feature extract issues
|
2023-06-14 14:46:22 -07:00 |
_helpers.py
|
…
|
|
_hub.py
|
More models w/ multi-weight support, moving to HF hub. Removing inplace_abn from all models including TResNet
|
2023-04-20 22:44:49 -07:00 |
_manipulate.py
|
…
|
|
_pretrained.py
|
Allow passing state_dict directly via pretrained cfg mechanism as an override
|
2023-05-08 15:15:44 -07:00 |
_prune.py
|
…
|
|
_registry.py
|
…
|
|
beit.py
|
Comment out beit url, no longer valid as now require long query string, leave for reference, must use HF hub now.
|
2023-08-03 10:00:46 -07:00 |
byoanet.py
|
Missed hub_id entries for byoanet models
|
2023-05-10 15:58:55 -07:00 |
byobnet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
cait.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
coat.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
convit.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
convmixer.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
convnext.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
crossvit.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
cspnet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
davit.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
deit.py
|
Tweak DinoV2 add, add MAE ViT weights, add initial intermediate layer getter experiment
|
2023-05-09 17:59:22 -07:00 |
densenet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
dla.py
|
Missed a few pretrained tags...
|
2023-05-10 12:16:30 -07:00 |
dpn.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
edgenext.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
efficientformer.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
efficientformer_v2.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
efficientnet.py
|
Clean more feature extract issues
|
2023-06-14 14:46:22 -07:00 |
eva.py
|
use torch.jit.Final instead of Final for beit, eva
|
2023-05-12 09:12:14 -07:00 |
factory.py
|
…
|
|
features.py
|
…
|
|
focalnet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
fx_features.py
|
…
|
|
gcvit.py
|
Missed a few pretrained tags...
|
2023-05-10 12:16:30 -07:00 |
ghostnet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
hardcorenas.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
helpers.py
|
…
|
|
hrnet.py
|
Clean more feature extract issues
|
2023-06-14 14:46:22 -07:00 |
hub.py
|
Update hub.py
|
2023-05-12 21:48:46 +02:00 |
inception_resnet_v2.py
|
Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights
|
2023-05-09 18:00:41 -07:00 |
inception_v3.py
|
Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights
|
2023-05-09 18:00:41 -07:00 |
inception_v4.py
|
Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights
|
2023-05-09 18:00:41 -07:00 |
levit.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
maxxvit.py
|
Add 21k maxvit tf weights
|
2023-05-10 18:23:32 -07:00 |
metaformer.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
mlp_mixer.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
mobilenetv3.py
|
Clean more feature extract issues
|
2023-06-14 14:46:22 -07:00 |
mobilevit.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
mvitv2.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
nasnet.py
|
Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights
|
2023-05-09 18:00:41 -07:00 |
nest.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
nfnet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
pit.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
pnasnet.py
|
Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights
|
2023-05-09 18:00:41 -07:00 |
pvt_v2.py
|
Missed a few pretrained tags...
|
2023-05-10 12:16:30 -07:00 |
registry.py
|
…
|
|
regnet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
repvit.py
|
Two small fixes, num_classes in base class, add model tag
|
2023-07-26 13:18:49 -07:00 |
res2net.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
resnest.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
resnet.py
|
Add proper pool size for new resnexts
|
2023-07-26 14:36:03 -07:00 |
resnetv2.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
rexnet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
selecsls.py
|
Fix unfortunate selecsls case bug caused by aggressive IDE rename
|
2023-08-03 10:37:06 -07:00 |
senet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
sequencer.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
sknet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
swin_transformer.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
swin_transformer_v2.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
swin_transformer_v2_cr.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
tnt.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
tresnet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
twins.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
vgg.py
|
Final push to get remaining models using multi-weight pretrained configs, almost all weights on HF hub
|
2023-04-26 15:52:13 -07:00 |
visformer.py
|
F.sdpa for visformer fails w/o contiguous on qkv, make experimental
|
2023-05-11 11:37:37 -07:00 |
vision_transformer.py
|
Fix ijepa vit issue with 448 model, minor formatting fixes
|
2023-07-26 20:46:27 -07:00 |
vision_transformer_hybrid.py
|
Tweak DinoV2 add, add MAE ViT weights, add initial intermediate layer getter experiment
|
2023-05-09 17:59:22 -07:00 |
vision_transformer_relpos.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
vision_transformer_sam.py
|
‘add ViT weight from I-JEPA pretrain’
|
2023-06-14 22:30:31 +08:00 |
volo.py
|
Replace deprecated NumPy aliases of builtin types
|
2023-07-03 22:24:25 +03:00 |
vovnet.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
xception.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
xception_aligned.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |
xcit.py
|
Add typing to all model entrypoint fns, add old cache check env var to builder
|
2023-05-08 08:52:38 -07:00 |