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
3196d6b131
Merge pull request #2230 from TianyiFranklinWang/main
...
Avoid zero division error
2024-07-14 21:35:24 -07:00
d3ce5a8665
Avoid zero division error
2024-07-15 12:45:46 +10:00
417cf7f871
Initialize weights of reg_token for ViT
2024-07-13 11:11:42 +08:00
f920119f3b
Fixing tests
2024-07-09 14:53:20 -07:00
644abf9588
Fix default_cfg test for mobilenet_100
2024-07-09 12:52:24 -07:00
d5afe106dc
Merge remote-tracking branch 'origin/tiny_test_models' into small_things
2024-07-09 12:49:57 -07:00
55101028bb
Rename test_tiny* -> test*. Fix ByobNet BasicBlock attn location and add test_byobnet model.
2024-07-09 11:53:11 -07:00
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
800405d941
Add conv_large mobilenetv3 aa/blur model defs
2024-07-08 13:50:05 -07:00
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
83c2c2f0c5
Add 'Maybe' PIL / image tensor conversions in case image alread in tensor format
2024-07-08 13:43:51 -07:00
648aaa4123
Merge pull request #2223 from stes/patch-1
...
Fix typo in type annotations in timm.models.hrnet
2024-07-08 07:34:47 -07:00
c01a47c9e7
Fix typo in type annotations in timm.models.hrnet
2024-07-08 00:53:16 +02:00
20fe56bd90
Merge pull request #2217 from dsuess/2216_fix_script_on_features_fx
...
Fix jit.script breaking with features_fx
2024-06-28 16:13:02 -07:00
197c10463b
Fix jit.script breaking with features_fx
2024-06-28 03:58:51 +00:00
d4ef0b4d58
Update README.md
2024-06-25 08:48:16 -07:00
b751da692d
Add latest ix (xavier init for mqa) hybrid medium & large weights for MobileNetV4
2024-06-24 13:49:55 -07:00
d4d4d84fda
Dev version 1.0.8.dev0
2024-06-24 11:34:13 -07:00
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
e7b4ab6a8d
Merge pull request #2214 from Sejik/main
...
Fix typo
2024-06-24 09:23:25 -07:00
c33a001397
Fix typo
2024-06-24 21:54:38 +09:00
02d0f27721
cleanup davit padding
2024-06-22 12:06:46 -07:00
c715c724e7
Fix tracing by removing float cast, should end up float anyways
2024-06-22 08:35:30 -07:00
fb58a73033
Fix #2212 map florence2 image tower to davit with a few changes
2024-06-21 15:31:29 -07:00
b28945ff05
Version 1.0.7, prep for release
2024-06-18 16:19:43 -07:00
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
427b3e46bd
Update README.md
2024-06-17 11:09:55 -07:00
16e082e1c2
Add mobilenetv4 hybrid-large weights
2024-06-17 11:08:31 -07:00
e41125cc83
Merge pull request #2209 from huggingface/fcossio-vit-maxpool
...
ViT pooling refactor
2024-06-17 07:51:12 -07:00
6254dfaece
Add numpy<2.0 to requirements until tests are sorted out for pytorch 2.3 vs older
2024-06-16 11:24:45 -07:00
a22466852d
Add 2400 epoch mobilenetv4 small weights, almost at paper, rounds to 73.8
2024-06-16 10:51:00 -07:00
b1a6f4a946
Some missed reset_classifier() type annotations
2024-06-16 10:39:27 -07:00
71101ebba0
Refactor vit pooling to add more reduction options, separately callable
2024-06-14 23:16:58 -07:00
a0bb5b4a44
Missing stem_kernel_size argument in EfficientNetFeatures
2024-06-14 13:39:31 -07:00
9567cf6d84
Feature: add option global_pool='max' to VisionTransformer
...
Most of the CNNs have a max global pooling option. I would like to extend ViT to have this option.
2024-06-14 15:24:54 +02:00
9613c76844
Add mobilenet edgetpu defs for exp, add ol mobilenet v1 back for completeness / comparison
2024-06-13 17:33:04 -07:00
22de845add
Prepping for final MobileCLIP weight locations ( #2199 )
...
* Prepping for final MobileCLIP weight locations
* Update weight locations to coreml-projects
* Update mobileclip weight locations with final apple org location
2024-06-13 16:55:49 -07:00
575978ba55
Add mnv4_conv_large 384x384 weight location
2024-06-13 12:58:04 -07:00
832d3618a5
Update README.md
2024-06-12 23:26:05 -07:00
7b5f17d1bd
Update README.md, bump dev version 1.0.6
2024-06-12 12:35:44 -07:00
e42e453128
Fix mmnv4 conv_large weight link, reorder mnv4 pretrained cfg for proper precedence
2024-06-12 11:16:49 -07:00
7b0a5321cb
Merge pull request #2198 from huggingface/openai_clip_resnet
...
Mapping OpenAI CLIP Modified ResNet weights -> ByobNet.
2024-06-12 09:33:30 -07:00
57adc1acc8
Fix rotary embed version of attn pool. Bit of cleanup/naming
2024-06-11 23:49:17 -07:00
5aa49d56bf
Merge pull request #2202 from huggingface/mnv4_first_weights
...
First set of MobileNetV4 weights trained in timm
2024-06-11 23:06:22 -07:00
cdc7bcea69
Make 2d attention pool modules compatible with head interface. Use attention pool in CLIP ResNets as head. Make separate set of GAP models w/ avg pool instead of attn pool.
2024-06-11 21:32:07 -07:00
c63da1405c
Pretrained cfg name mismatch
2024-06-11 21:16:54 -07:00
88efca1be2
First set of MobileNetV4 weights trained in timm
2024-06-11 18:53:01 -07:00
30ffa152de
Fix load of larger ResNet CLIP models, experimenting with making AttentionPool *the* head, seems to fine-tune better, one less layer.
2024-06-10 12:07:14 -07:00
5e9ff5798f
Adding pos embed resize fns to FX autowrap exceptions
2024-06-10 12:06:47 -07:00
Ross Wightman
Ross Wightman
Tianyi Wang
Promisery
Steffen Schneider
Daniel Suess
Sejik
Fernando Cossio