146c2fbe34
Add resnet50d and efficientnet_b1 ra4 (mnv4) hparam weights
2024-08-16 12:10:00 -07:00
962958723c
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
f2cfb4c677
Add WIP HieraDet impl (SAM2 backbone support)
2024-08-15 17:58:15 -07:00
a50e53d41f
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
2f3fed43b8
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
fee91fdd41
Update Hiera model for abswin, more stable weight init, layer-scale. ImageNet-12k weights for hiera_small_abswin, and two of the sbb vits with improved reg4 init.
2024-08-14 12:22:40 -07:00
ac3470188b
Add mobilenetv1_125 weights from ra4 / mnv4 hparam series
2024-08-14 08:47:02 -07:00
907555fda6
Update model img-size/crop expansion for bulk runner
2024-08-13 10:42:09 -07:00
4515a435e4
Update imagenet & OOD test set eval results with current weights. Now have both test & train resolution results in tables.
2024-08-13 10:25:25 -07:00
2df9f2869d
Merge pull request #2257 from huggingface/weights_only
...
Add weights only flag to avoid warning, try to keep bwd compat.
2024-08-12 18:09:36 -07:00
a7b0bfc376
exception for unknown arg should have been TypeError
2024-08-12 16:09:36 -07:00
bd0f79153b
Add weights only flag to avoid warning, try to keep bwd compat. Default to True for remote load of pretrained weights, keep False for local checkpoing load to avoid training checkpoint breaks.. fix #2249
2024-08-12 11:41:41 -07:00
531215eded
Bring changelog up to current, back to dev version, 1.0.9.dev0
2024-08-08 09:17:23 -07:00
d29b720d8d
Add changes to hfdocs toctree
2024-08-08 09:12:58 -07:00
4fb878bdd6
Remove old docs
2024-08-08 09:08:56 -07:00
f96e369615
Update changes.md
2024-08-08 08:37:49 -07:00
33ab2b35ae
Update README.md
2024-08-08 08:36:04 -07:00
bf781bc3c2
Merge pull request #2245 from dhkim0225/feature/rdnet
...
Feature: Add RDNet
2024-08-07 16:19:22 -07:00
49f6e1da96
Remove eval to avoid safety concerns
2024-08-07 14:54:13 -07:00
2d5c9bf60d
Merge pull request #2253 from brendanartley/hgnet-grad-checkpointing
...
Add gradient checkpointing to hgnets
2024-08-07 12:45:14 -07:00
e9ef9424f0
Add a few missing __all__ entries.
2024-08-07 09:35:51 -07:00
232a459e36
Added gradient checkpointing to hgnet
2024-08-07 09:22:51 -07:00
ea33d6a74d
bugfix pool_size
2024-08-03 09:50:20 +09:00
10344625be
proj_drop wasn't being called in sam vit
2024-08-02 16:12:02 -07:00
6fdc9d8243
update code for ci pass
2024-08-02 15:14:40 +09:00
7a866b6521
update code for torchscript
2024-08-02 09:58:13 +09:00
3b6a1d4d48
model summary added
2024-08-02 09:11:44 +09:00
8248122f82
add rdnet
2024-08-01 14:54:29 +09:00
025259024d
add rdnet
2024-08-01 14:51:15 +09:00
225f4f92b3
add rdnet
2024-08-01 14:49:21 +09:00
a6fe31b096
Update README.md
2024-07-28 22:16:06 -07:00
4a10302754
Add mobilenet_edgetpu_v2_m weights
2024-07-28 17:19:36 -07:00
70182ea0cd
Version 1.0.8
2024-07-26 17:05:36 -07:00
ab8cb070fc
Add xavier_uniform init of MNVC hybrid attention modules. Small improvement in training stability.
2024-07-26 17:03:40 -07:00
9558a7f8eb
Update README.md
2024-07-26 10:16:44 -07:00
cec70b6779
Merge pull request #2225 from huggingface/small_things
...
Small things
2024-07-25 20:29:13 -07:00
61df3fde89
Wrong hybrid_medium in12k pool sizes
2024-07-25 15:39:21 -07:00
9aa2930760
Add latest mobilenetv4 and baseline updates for mobilenetv1 and efficientnet_b0 weights
2024-07-25 14:20:54 -07:00
7b6a406474
remove swin debug prints
2024-07-24 21:05:56 -07:00
4c531be479
set_input_size(), always_partition, strict_img_size, dynamic mask option for all swin models. More flexibility in resolution, window resizing.
2024-07-24 16:41:31 -07:00
8b14fc7bb6
Merge pull request #2240 from Zirunis/Zirunis-patch-1
...
Fix LR scheduler help in train.py
2024-07-23 11:04:18 -07:00
f3c11dc3a5
Merge pull request #2239 from huggingface/fix_out_indices_order
...
Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet
2024-07-23 11:02:42 -07:00
8efdc38213
Fix #2242 add checks for out indices with intermediate getter mode
2024-07-23 08:19:09 -07:00
4ed93fce93
Fix LR scheduler help in train.py
...
The default is and always has been the cosine scheduler, yet the help states that the default would be the step scheduler. Whatever the intended one was, for backwards compatibility the default should definitely remain cosine, which is why I changed the help comment to reflect that.
2024-07-22 23:04:00 +02:00
d2240745d3
Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set()
2024-07-22 13:33:30 -07:00
2b3f1a4633
Make channels for classic resnet configurable
2024-07-22 10:47:40 -07:00
9b2b8014e8
Add weights for test models
2024-07-22 10:08:57 -07:00
a1996ec0f4
Merge pull request #2238 from huggingface/fix_mnv4_query_strides
...
Fix mnv4 query strides
2024-07-19 16:32:08 -07:00
7e0caa1ba3
Padding helpers work if tuples/lists passed
2024-07-19 14:28:03 -07:00
2180800646
MQA query_strides bugs fix #2237 . No padding for avg_pool2d if not 'same', use scale_factor for Upsample.
2024-07-19 14:26:54 -07:00
Ross Wightman
Ross Wightman
brendanartley
dong-hyun
Zirunis