Ross Wightman
|
392368e210
|
Add efficientnetv2_rw_t defs w/ weights, and gc variant, as well as gcresnet26ts for experiments. Version 0.4.13
|
2021-07-09 16:46:52 -07:00 |
|
Ross Wightman
|
d17b374f0f
|
Minimum input_size needed to be higher
|
2021-06-08 21:31:39 -07:00 |
|
Ross Wightman
|
b3b90d944d
|
Add min_input_size to bat_resnext to prevent test breakage.
|
2021-06-08 17:32:08 -07:00 |
|
Ross Wightman
|
10d8fa4620
|
Add gc and bat attention resnext26ts variants to byob for test.
|
2021-06-08 14:21:07 -07:00 |
|
Ross Wightman
|
307a935b79
|
Add non-local and BAT attention. Merge attn and self-attn factories into one. Add attention references to README. Add mlp 'mode' to ECA.
|
2021-05-31 13:18:11 -07:00 |
|
Ross Wightman
|
9611458e19
|
Throw in some FBNetV3 code I had lying around, some refactoring of SE reduction channel calcs for all EffNet archs.
|
2021-05-28 20:47:24 -07:00 |
|
Ross Wightman
|
02f9d4bc34
|
Add weights for resnet51q model, add 61q def.
|
2021-05-28 09:53:16 -07:00 |
|
Ross Wightman
|
f615474be3
|
Fix broken test, repvgg block doesn't have attn_last attr.
|
2021-05-27 18:12:22 -07:00 |
|
Ross Wightman
|
742c2d5247
|
Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.
|
2021-05-27 18:03:29 -07:00 |
|
Ross Wightman
|
9a3ae97311
|
Another set of byoanet models w/ ECA channel + SA + groups
|
2021-05-15 22:56:51 -07:00 |
|
Ross Wightman
|
b3d7580df1
|
Update ByoaNet comments. Fix first Steam feat chs for ByobNet.
|
2021-04-12 12:11:35 -07:00 |
|
Ross Wightman
|
7c97e66f7c
|
Remove commented code, add more consistent seed fn
|
2021-04-12 09:51:36 -07:00 |
|
Ross Wightman
|
ce62f96d4d
|
ByoaNet with bottleneck transformer, lambda resnet, and halo net experiments
|
2021-04-12 09:38:02 -07:00 |
|
Ross Wightman
|
6853b07bbd
|
Improve RegVGG block identity/vs non for clariy and fix attn usage. Add comments.
|
2021-02-10 14:40:29 -08:00 |
|
Ross Wightman
|
dc85e5a237
|
Add ByobNet w/ GPU-EfficientNets and RepVGG. Also add classic vgg models.
|
2021-02-09 16:22:52 -08:00 |