mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
pytorch-image-models/timm/layers
History
Adam J. Stewart f5c4d5cbb7
Add missing imports
2025-01-11 15:13:16 +01:00
..
__init__.py
Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override
2025-01-06 11:28:39 -08:00
activations_me.py
…
activations.py
…
adaptive_avgmax_pool.py
Stricter check on pool_type for adaptive pooling module. Fix #2159
2024-05-03 16:16:51 -07:00
attention2d.py
Add missing imports
2025-01-11 15:13:16 +01:00
attention_pool2d.py
Fix rotary embed version of attn pool. Bit of cleanup/naming
2024-06-11 23:49:17 -07:00
attention_pool.py
Mapping OpenAI CLIP Modified ResNet weights -> ByobNet. Improve AttentionPool2d layers. Fix #1731
2024-06-09 16:54:48 -07:00
blur_pool.py
Add anti-aliasing support to mobilenetv3 and efficientnet family models. Update MobileNetV4 model defs, resolutions. Fix #599
2024-05-27 22:06:22 -07:00
bottleneck_attn.py
…
cbam.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
classifier.py
More Hiera updates. Add forward_intermediates to hieradat/sam2 impl. Make both use same classifier module. Add coarse bool to intermediates.
2024-08-16 11:10:04 -07:00
cond_conv2d.py
…
config.py
Update README, few minor fixups.
2025-01-06 13:09:15 -08:00
conv2d_same.py
Add onnx utils and export code, tweak padding and conv2d_same for better dynamic export with recent PyTorch
2023-04-11 17:03:57 -07:00
conv_bn_act.py
See if we can avoid some model / layer pickle issues with the aa attr in ConvNormAct
2024-12-03 08:02:55 -08:00
create_act.py
Add WIP HieraDet impl (SAM2 backbone support)
2024-08-15 17:58:15 -07:00
create_attn.py
…
create_conv2d.py
…
create_norm_act.py
Consistency handling None / empty string inputs to norm / act create fns
2023-11-03 11:01:41 -07:00
create_norm.py
Add SimpleNorm to create_norm factory
2024-12-30 19:24:21 -08:00
drop.py
Fix meshgrid deprecation warnings and backward compat with explicit 'ndgrid' and 'meshgrid' fn w/o indexing arg
2024-01-27 13:48:33 -08:00
eca.py
…
evo_norm.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
fast_norm.py
Use torch F.rms_norm when possible, select fast vs normal paths appropriately and test with torchscript
2024-12-30 19:24:21 -08:00
filter_response_norm.py
…
format.py
…
gather_excite.py
…
global_context.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
grid.py
Fix meshgrid deprecation warnings and backward compat with explicit 'ndgrid' and 'meshgrid' fn w/o indexing arg
2024-01-27 13:48:33 -08:00
grn.py
Add ConvNeXt-V2 support (model additions and weights) (#1614)
2023-01-05 07:53:32 -08:00
halo_attn.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
helpers.py
Fixed Typos
2023-10-21 21:46:31 -07:00
hybrid_embed.py
set_input_size initial impl for vit & swin v1. Move HybridEmbed to own location in timm/layers
2024-07-17 15:25:48 -07:00
inplace_abn.py
…
interpolate.py
Add more advanced interpolation method from BEiT and support non-square window & image size adaptation for
2023-08-08 16:41:16 -07:00
lambda_layer.py
…
layer_scale.py
Fix hiera init with num_classes=0, fix weight tag names for sbb2 hiera/vit weights, add LayerScale/LayerScale2d to layers
2024-08-15 11:14:38 -07:00
linear.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
median_pool.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
mixed_conv2d.py
…
ml_decoder.py
Remove JIT activations, take jit out of ME activations. Remove other instances of torch.jit.script. Breaks torch.compile and is much less performant. Remove SpaceToDepthModule
2024-05-06 16:32:49 -07:00
mlp.py
Add comments to MLP layers re expected layouts
2025-01-02 09:41:35 -08:00
non_local_attn.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
norm_act.py
Add ConvStem and MobileCLIP hybrid model for B variant. Add full norm disable support to ConvNormAct layers
2024-06-06 09:15:27 -07:00
norm.py
Use torch F.rms_norm when possible, select fast vs normal paths appropriately and test with torchscript
2024-12-30 19:24:21 -08:00
padding.py
…
patch_dropout.py
…
patch_embed.py
set_input_size initial impl for vit & swin v1. Move HybridEmbed to own location in timm/layers
2024-07-17 15:25:48 -07:00
pool2d_same.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
pos_embed_rel.py
…
pos_embed_sincos.py
Mapping OpenAI CLIP Modified ResNet weights -> ByobNet. Improve AttentionPool2d layers. Fix #1731
2024-06-09 16:54:48 -07:00
pos_embed.py
…
selective_kernel.py
Remove separate ConvNormActAa class, merge with ConvNormAct
2024-06-10 12:05:35 -07:00
separable_conv.py
…
space_to_depth.py
Remove JIT activations, take jit out of ME activations. Remove other instances of torch.jit.script. Breaks torch.compile and is much less performant. Remove SpaceToDepthModule
2024-05-06 16:32:49 -07:00
split_attn.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
split_batchnorm.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
squeeze_excite.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
std_conv.py
…
test_time_pool.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
trace_utils.py
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
typing.py
include suggestions from review
2023-10-30 13:47:54 -07:00
weight_init.py
Rename global pos embed for Hiera abswin, factor out commonly used vit weight init fns to layers. Add a channels-last ver of normmlp head.
2024-08-15 17:46:36 -07:00