Ross Wightman
|
e9ef9424f0
|
Add a few missing __all__ entries.
|
2024-08-07 09:35:51 -07:00 |
|
brendanartley
|
232a459e36
|
Added gradient checkpointing to hgnet
|
2024-08-07 09:22:51 -07:00 |
|
dong-hyun
|
ea33d6a74d
|
bugfix pool_size
|
2024-08-03 09:50:20 +09:00 |
|
Ross Wightman
|
10344625be
|
proj_drop wasn't being called in sam vit
|
2024-08-02 16:12:02 -07:00 |
|
dong-hyun
|
6fdc9d8243
|
update code for ci pass
|
2024-08-02 15:14:40 +09:00 |
|
dong-hyun
|
7a866b6521
|
update code for torchscript
|
2024-08-02 09:58:13 +09:00 |
|
dong-hyun
|
8248122f82
|
add rdnet
|
2024-08-01 14:54:29 +09:00 |
|
dong-hyun
|
025259024d
|
add rdnet
|
2024-08-01 14:51:15 +09:00 |
|
dong-hyun
|
225f4f92b3
|
add rdnet
|
2024-08-01 14:49:21 +09:00 |
|
Ross Wightman
|
4a10302754
|
Add mobilenet_edgetpu_v2_m weights
|
2024-07-28 17:19:36 -07:00 |
|
Ross Wightman
|
70182ea0cd
|
Version 1.0.8
|
2024-07-26 17:05:36 -07:00 |
|
Ross Wightman
|
ab8cb070fc
|
Add xavier_uniform init of MNVC hybrid attention modules. Small improvement in training stability.
|
2024-07-26 17:03:40 -07:00 |
|
Ross Wightman
|
cec70b6779
|
Merge pull request #2225 from huggingface/small_things
Small things
|
2024-07-25 20:29:13 -07:00 |
|
Ross Wightman
|
61df3fde89
|
Wrong hybrid_medium in12k pool sizes
|
2024-07-25 15:39:21 -07:00 |
|
Ross Wightman
|
9aa2930760
|
Add latest mobilenetv4 and baseline updates for mobilenetv1 and efficientnet_b0 weights
|
2024-07-25 14:20:54 -07:00 |
|
Ross Wightman
|
7b6a406474
|
remove swin debug prints
|
2024-07-24 21:05:56 -07:00 |
|
Ross Wightman
|
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 |
|
Ross Wightman
|
8efdc38213
|
Fix #2242 add checks for out indices with intermediate getter mode
|
2024-07-23 08:19:09 -07:00 |
|
Ross Wightman
|
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 |
|
Ross Wightman
|
2b3f1a4633
|
Make channels for classic resnet configurable
|
2024-07-22 10:47:40 -07:00 |
|
Ross Wightman
|
9b2b8014e8
|
Add weights for test models
|
2024-07-22 10:08:57 -07:00 |
|
Ross Wightman
|
7e0caa1ba3
|
Padding helpers work if tuples/lists passed
|
2024-07-19 14:28:03 -07:00 |
|
Ross Wightman
|
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
|
1a05ed29a1
|
Add to 'abswin' hiera models for train trials
|
2024-07-19 11:05:31 -07:00 |
|
Ross Wightman
|
0cbf4fa586
|
_orig_mod still causing issues even though I thought it was fixed in pytorch, add unwrap / clean helpers
|
2024-07-19 11:03:45 -07:00 |
|
Feraidoon Mehri
|
4cca568bd8
|
eva.py: fixed bug in applying attention mask
The mask should be applied before the softmax.
|
2024-07-19 15:12:04 +03:30 |
|
Ross Wightman
|
3a8a965891
|
Implement absolute+window pos embed for hiera, resizable but needs new weights
|
2024-07-18 21:43:37 -07:00 |
|
Ross Wightman
|
7160af4a24
|
Merge pull request #2229 from Promisery/reg_token
Initialize weights of reg_token for ViT
|
2024-07-18 09:25:29 -07:00 |
|
Ross Wightman
|
392b78aee7
|
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 |
|
Ross Wightman
|
34c9fee554
|
Fix pass through of input / target keys so ImageDataset readers so args work with hfds instead of just hfids (iterable)
|
2024-07-17 10:11:46 -07:00 |
|
Promisery
|
417cf7f871
|
Initialize weights of reg_token for ViT
|
2024-07-13 11:11:42 +08:00 |
|
Ross Wightman
|
f920119f3b
|
Fixing tests
|
2024-07-09 14:53:20 -07:00 |
|
Ross Wightman
|
644abf9588
|
Fix default_cfg test for mobilenet_100
|
2024-07-09 12:52:24 -07:00 |
|
Ross Wightman
|
d5afe106dc
|
Merge remote-tracking branch 'origin/tiny_test_models' into small_things
|
2024-07-09 12:49:57 -07:00 |
|
Ross Wightman
|
55101028bb
|
Rename test_tiny* -> test*. Fix ByobNet BasicBlock attn location and add test_byobnet model.
|
2024-07-09 11:53:11 -07:00 |
|
Ross Wightman
|
1334598462
|
Add support back to EfficientNet to disable head_conv / bn2 so mobilnetv1 can be implemented properly
|
2024-07-08 13:51:26 -07:00 |
|
Ross Wightman
|
800405d941
|
Add conv_large mobilenetv3 aa/blur model defs
|
2024-07-08 13:50:05 -07:00 |
|
Ross Wightman
|
f81b094aaa
|
Add 'qkv_bias_separate' flag for EVA/beit/swinv2 attn modules to allow an override for easy quantization wrappers. Fix #2098
|
2024-07-08 13:48:38 -07:00 |
|
Ross Wightman
|
83c2c2f0c5
|
Add 'Maybe' PIL / image tensor conversions in case image alread in tensor format
|
2024-07-08 13:43:51 -07:00 |
|
Steffen Schneider
|
c01a47c9e7
|
Fix typo in type annotations in timm.models.hrnet
|
2024-07-08 00:53:16 +02:00 |
|
Daniel Suess
|
197c10463b
|
Fix jit.script breaking with features_fx
|
2024-06-28 03:58:51 +00:00 |
|
Ross Wightman
|
b751da692d
|
Add latest ix (xavier init for mqa) hybrid medium & large weights for MobileNetV4
|
2024-06-24 13:49:55 -07:00 |
|
Ross Wightman
|
d4d4d84fda
|
Dev version 1.0.8.dev0
|
2024-06-24 11:34:13 -07:00 |
|
Ross Wightman
|
f8342a045a
|
Merge pull request #2213 from huggingface/florence2
Fix #2212 map florence2 image tower to davit with a few changes
|
2024-06-24 11:01:08 -07:00 |
|
Sejik
|
c33a001397
|
Fix typo
|
2024-06-24 21:54:38 +09:00 |
|
Ross Wightman
|
02d0f27721
|
cleanup davit padding
|
2024-06-22 12:06:46 -07:00 |
|
Ross Wightman
|
c715c724e7
|
Fix tracing by removing float cast, should end up float anyways
|
2024-06-22 08:35:30 -07:00 |
|
Ross Wightman
|
fb58a73033
|
Fix #2212 map florence2 image tower to davit with a few changes
|
2024-06-21 15:31:29 -07:00 |
|
Ross Wightman
|
b28945ff05
|
Version 1.0.7, prep for release
|
2024-06-18 16:19:43 -07:00 |
|
Ross Wightman
|
fb13e6385e
|
Merge pull request #2203 from huggingface/more_mobile
Add mobilenet edgetpu defs for exp, add ol mobilenet v1 back for comp…
|
2024-06-18 15:20:01 -07:00 |
|