Ross Wightman 7c7ecd2492 Add --use-train-size flag to force use of train input_size (over test input size) for validation. Default test-time pooling to use train input size (fixes issues).

2022-07-07 22:01:24 -07:00

..

__init__.py

Add MobileVitV2 support. Fix #1332. Move GroupNorm1 to common layers (used in poolformer + mobilevitv2). Keep ol custom ConvNeXt LayerNorm2d impl as LayerNormExp2d for reference.

2022-07-07 14:41:01 -07:00

activations_jit.py

…

activations_me.py

…

activations.py

…

adaptive_avgmax_pool.py

…

attention_pool2d.py

…

blur_pool.py

Make a few more layers symbolically traceable (remove from FX leaf modules)

2022-03-24 21:43:56 -07:00

bottleneck_attn.py

…

cbam.py

…

classifier.py

…

cond_conv2d.py

fixing channels_last on cond_conv2d; update nvfuser debug env variable

2022-04-25 12:41:46 -07:00

config.py

…

conv2d_same.py

…

conv_bn_act.py

Add more experimental darknet and 'cs2' darknet variants (different cross stage setup, closer to newer YOLO backbones) for train trials.

2022-07-01 15:14:01 -07:00

create_act.py

…

create_attn.py

Add MobileVitV2 support. Fix #1332. Move GroupNorm1 to common layers (used in poolformer + mobilevitv2). Keep ol custom ConvNeXt LayerNorm2d impl as LayerNormExp2d for reference.

2022-07-07 14:41:01 -07:00

create_conv2d.py

…

create_norm_act.py

…

drop.py

Improve repr for DropPath module

2022-06-24 14:58:15 -07:00

eca.py

…

evo_norm.py

Remove SiLU layer in default args that breaks import on old old PyTorch

2022-06-24 14:56:58 -07:00

filter_response_norm.py

…

gather_excite.py

…

global_context.py

…

halo_attn.py

…

helpers.py

…

inplace_abn.py

…

lambda_layer.py

…

linear.py

…

median_pool.py

…

mixed_conv2d.py

…

ml_decoder.py

…

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

2022-05-13 13:53:57 -07:00

non_local_attn.py

…

norm_act.py

Remove unecessary code for synbn guard

2022-06-24 14:57:42 -07:00

norm.py

Add MobileVitV2 support. Fix #1332. Move GroupNorm1 to common layers (used in poolformer + mobilevitv2). Keep ol custom ConvNeXt LayerNorm2d impl as LayerNormExp2d for reference.

2022-07-07 14:41:01 -07:00

padding.py

…

patch_embed.py

…

pool2d_same.py

…

pos_embed.py

…

selective_kernel.py

…

separable_conv.py

…

space_to_depth.py

…

split_attn.py

…

split_batchnorm.py

…

squeeze_excite.py

…

std_conv.py

…

test_time_pool.py

Add --use-train-size flag to force use of train input_size (over test input size) for validation. Default test-time pooling to use train input size (fixes issues).

2022-07-07 22:01:24 -07:00

trace_utils.py

…

weight_init.py

Add eps arg to LayerNorm2d, add 'tf' (tensorflow) variant of trunc_normal_ that applies scale/shift after sampling (instead of needing to move a/b)

2022-07-01 15:15:39 -07:00