This website works better with JavaScript.
Explore
Help
Register
Sign In
Migrations
/
pytorch-image-models
Watch
1
Star
0
Fork
You've already forked pytorch-image-models
0
Code
Issues
Pull Requests
Packages
Projects
Releases
Wiki
Activity
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
a050fde5cd
main
focalnet_and_swin_refactor
fix_window_trace
scaling_vit
dataset_info
levit_efficientformer_redux
norm_mlp_classifier
0.6.x
maxxvit_hub
freeze_norm_revisit
parse_kwargs2
parse_kwargs
gh-pages
convnextv2
multi-weight_effnet_convnext
refactor-imports
eva
multi-weight
update-test-pytorch
bits_and_tpu
script_cleanup
clip_laion2b
more_vit
fix_tests
edgenext_csp_and_more
fixes-syncbn_pretrain_cfg_resolve
swin_v2
vit_relpos_refactor
norm_norm_norm
0.5.x
convnext_and_copyright
more_datasets
fixes_bce_regnet
attn_update
perceiver
vit_and_bit_test_fixes
cleanup_xla_model_fixes
more_attn
hf_hub_revisit
xsepconv
efficientnet_attn
v0.1-attn-weights
v0.1-cadene
v0.1-coat-weights
v0.1-dnf-weights
v0.1-effv2-weights
v0.1-ger-weights
v0.1-hrnet
v0.1-mvit-weights
v0.1-pit-weights
v0.1-regnet
v0.1-repvgg-weights
v0.1-res2net
v0.1-resnest
v0.1-rexnet
v0.1-rs-weights
v0.1-rsb-weights
v0.1-selecsls
v0.1-tpu-weights
v0.1-tresnet
v0.1-vitjx
v0.1-vt3p-weights
v0.1-weights
v0.1-weights-maxx
v0.1-weights-morevit
v0.1-weights-swinv2
v0.1.1
v0.2.1
v0.3.3
v0.3.4
v0.4.12
v0.4.5
v0.4.9
v0.5.4
v0.6.11
v0.6.12
v0.6.5
v0.6.7
v0.8.10dev0
v0.8.13dev0
v0.8.2dev0
v0.8.6dev0
Branches
Tags
${ item.name }
Create tag
${ searchTerm }
Create branch
${ searchTerm }
from 'a050fde5cd'
${ noResults }
pytorch-image-models
/
timm
/
models
/
layers
History
Ross Wightman
07d0c4ae96
Improve repr for DropPath module
3 years ago
..
__init__.py
Support BatchNormAct2d for sync-bn use.
Fix
#1254
3 years ago
activations.py
…
activations_jit.py
…
activations_me.py
…
adaptive_avgmax_pool.py
…
attention_pool2d.py
Significant model refactor and additions:
3 years ago
blur_pool.py
Make a few more layers symbolically traceable (remove from FX leaf modules)
3 years ago
bottleneck_attn.py
wip - pre-rebase
3 years ago
cbam.py
Significant norm update
3 years ago
classifier.py
Significant model refactor and additions:
3 years ago
cond_conv2d.py
fixing channels_last on cond_conv2d; update nvfuser debug env variable
3 years ago
config.py
…
conv2d_same.py
…
conv_bn_act.py
Significant norm update
3 years ago
create_act.py
Updated EvoNorm implementations with some experimentation. Add FilterResponseNorm. Updated RegnetZ and ResNetV2 model defs for trials.
3 years ago
create_attn.py
…
create_conv2d.py
Significant norm update
3 years ago
create_norm_act.py
Significant norm update
3 years ago
drop.py
Improve repr for DropPath module
3 years ago
eca.py
…
evo_norm.py
Remove SiLU layer in default args that breaks import on old old PyTorch
3 years ago
filter_response_norm.py
Updated EvoNorm implementations with some experimentation. Add FilterResponseNorm. Updated RegnetZ and ResNetV2 model defs for trials.
3 years ago
gather_excite.py
…
global_context.py
wip - pre-rebase
3 years ago
halo_attn.py
wip - pre-rebase
3 years ago
helpers.py
…
inplace_abn.py
Significant norm update
3 years ago
lambda_layer.py
wip - pre-rebase
3 years ago
linear.py
…
median_pool.py
…
mixed_conv2d.py
…
ml_decoder.py
support CNNs
3 years ago
mlp.py
Add updated vit_relpos weights, and impl w/ support for official swin-v2 differences for relpos. Add bias control support for MLP layers
3 years ago
non_local_attn.py
Significant norm update
3 years ago
norm.py
Add ResNet-50 w/ GN (resnet50_gn) and SEBotNet-33-TS (sebotnet33ts_256) model defs and weights. Update halonet50ts weights w/ slightly better variant in1k val, more robust to test sets.
3 years ago
norm_act.py
Remove unecessary code for synbn guard
3 years ago
padding.py
…
patch_embed.py
wip - attempting to rebase
3 years ago
pool2d_same.py
…
pos_embed.py
Significant model refactor and additions:
3 years ago
selective_kernel.py
Significant norm update
3 years ago
separable_conv.py
Significant norm update
3 years ago
space_to_depth.py
…
split_attn.py
Significant norm update
3 years ago
split_batchnorm.py
…
squeeze_excite.py
…
std_conv.py
…
test_time_pool.py
…
trace_utils.py
Better fix for
#954
that doesn't break torchscript, pull torch._assert into timm namespace when it exists
3 years ago
weight_init.py
…