pytorch-image-models/timm/layers
方曦 170a5b6e27 add tinyvit 2023-09-01 11:05:56 -07:00
..
__init__.py add tinyvit 2023-09-01 11:05:56 -07:00
activations.py
activations_jit.py
activations_me.py
adaptive_avgmax_pool.py
attention_pool2d.py
blur_pool.py
bottleneck_attn.py
cbam.py
classifier.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
cond_conv2d.py
config.py
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 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
create_act.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
create_attn.py
create_conv2d.py
create_norm.py
create_norm_act.py
drop.py
eca.py
evo_norm.py
fast_norm.py Metaformer baselines for vision (final PR with cleanup) (#1793) 2023-05-05 11:18:26 -07:00
filter_response_norm.py
format.py
gather_excite.py
global_context.py
grn.py
halo_attn.py
helpers.py
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
linear.py
median_pool.py
mixed_conv2d.py
ml_decoder.py
mlp.py Fix missing norm call in Mlp forward (not used by default, but can be enabled for normformer MLP scale). Fix #1851 fix #1852 2023-08-03 11:36:30 -07:00
non_local_attn.py
norm.py
norm_act.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
padding.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
patch_dropout.py
patch_embed.py dynamic_size -> dynamic_img_size, add dynamic_img_pad for padding option 2023-08-27 15:58:35 -07:00
pool2d_same.py
pos_embed.py add tinyvit 2023-09-01 11:05:56 -07:00
pos_embed_rel.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
pos_embed_sincos.py Support dynamic_resize in eva.py models 2023-08-27 15:58:35 -07:00
selective_kernel.py
separable_conv.py
space_to_depth.py Always some torchscript issues 2023-04-26 20:42:34 -07:00
split_attn.py
split_batchnorm.py
squeeze_excite.py
std_conv.py
test_time_pool.py
trace_utils.py
weight_init.py