3cc4d7a894
Fix missing register for 224 eva model
2022-12-07 08:54:06 -08:00
eba07b0de7
Add eva models to beit.py
2022-12-07 08:54:06 -08:00
927f031293
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
3785c234d7
Remove clip vit models that won't be ft and comment two that aren't uploaded yet
2022-12-05 10:21:34 -08:00
f82239b30e
multi-weight branch version -> 0.8.0dev
2022-12-05 10:21:34 -08:00
755570e2d6
Rename _pretrained.py -> pretrained.py, not feasible to change the other files to same scheme without breaking uses
2022-12-05 10:21:34 -08:00
72cfa57761
Add ported Tensorflow MaxVit weights. Add a few more CLIP ViT fine-tunes. Tweak some model tag names. Improve model tag name sorting. Update HF hub push config layout.
2022-12-05 10:21:34 -08:00
4d5c395160
MaxVit, ViT, ConvNeXt, and EfficientNet-v2 updates
...
* Add support for TF weights and modelling specifics to MaxVit (testing ported weights)
* More fine-tuned CLIP ViT configs
* ConvNeXt and MaxVit updated to new pretrained cfgs use
* EfficientNetV2, MaxVit and ConvNeXt high res models use squash crop/resize
2022-12-05 10:21:34 -08:00
3db4e346e0
Switch TFDS dataset to use INTEGER_ACCURATE jpeg decode by default
2022-12-05 10:21:34 -08:00
9da7e3a799
Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate
2022-12-05 10:21:34 -08:00
b2b6285af7
Add two more FT clip weights
2022-12-05 10:21:34 -08:00
5895056dc4
Add openai b32 ft
2022-12-05 10:21:34 -08:00
9dea5143d5
Adding more clip ft variants
2022-12-05 10:21:34 -08:00
444dcba4ad
CLIP B16 12k weights added
2022-12-05 10:21:34 -08:00
dff4717cbf
Add clip b16 384x384 finetunes
2022-12-05 10:21:34 -08:00
883fa2eeaa
Add fine-tuned B/16 224x224 in1k clip models
2022-12-05 10:21:34 -08:00
9a3d2ac2d5
Add latest CLIP ViT fine-tune pretrained configs / model entrypt updates
2022-12-05 10:21:34 -08:00
42bbbddee9
Add missing model config
2022-12-05 10:21:34 -08:00
def68befa7
Updating vit model defs for mult-weight support trial (vit first). Prepping for CLIP (laion2b and openai) fine-tuned weights.
2022-12-05 10:21:34 -08:00
0dadb4a6e9
Initial multi-weight support, handled so old pretraing config handling co-exists with new tags.
2022-12-05 10:21:34 -08:00
653bdc7105
Fix comment: https://github.com/rwightman/pytorch-image-models/pull/1564#issuecomment-1326743424
2022-11-25 09:52:52 +08:00
bdc9fad638
Fix compatible BUG: QMNIST and ImageNet datasets do not exist in torchvision 0.10.1.
2022-11-24 14:37:44 +08:00
9b114754db
refactor push_to_hub helper
2022-11-16 12:03:34 +01:00
ae0a0db7de
Create repo before cloning with Repository.clone_from
2022-11-15 15:17:20 +01:00
803254bb40
Fix spacing misalignment for fast norm path in LayerNorm modules
2022-10-24 21:43:49 -07:00
475ecdfa3d
cast env var args for dataset readers to int
2022-10-17 14:40:11 -07:00
39190f5f44
Remove inplace operators when calculating the loss
...
Remove inplace operators to overcome the following error when using `asymmetric_loss`
```
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation
```
2022-10-17 10:41:21 +02:00
6635bc3f7d
Merge pull request #1479 from rwightman/script_cleanup
...
Train / val script enhancements, non-GPU (ie CPU) device support, HF datasets support, TFDS/WDS dataloading improvements
2022-10-15 09:29:39 -07:00
0e6023f032
Merge pull request #1381 from ChristophReich1996/master
...
Fix typo in PositionalEncodingFourier
2022-10-14 18:34:33 -07:00
66f4af7090
Merge remote-tracking branch 'origin/master' into script_cleanup
2022-10-14 15:54:00 -07:00
d3961536c9
comment some debug logs for WDS dataset
2022-10-14 15:39:00 -07:00
e9dccc918c
Rename dataset/parsers -> dataset/readers, create_parser to create_reader, etc
2022-10-14 15:14:38 -07:00
8c28363dc9
Version 0.7.dev0 for master
2022-10-14 09:38:02 -07:00
30bafd7347
🔖 add dev suffix to version tag
2022-10-13 17:08:33 -04:00
f67a7ee8bd
Set num_workers in Iterable WDS/TFDS datasets early so sample estimate is correct
2022-10-11 15:11:18 -07:00
cea8df3d0c
Version 0.6.12
2022-10-10 21:49:52 -07:00
9914f744dc
Add more maxxvit weights includ ConvNeXt conv block based experiments.
2022-10-10 21:49:18 -07:00
b1b024dfed
Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168
2022-10-07 10:43:04 -07:00
4f18d6dc5f
Fix logs in WDS parser
2022-10-07 10:06:17 -07:00
8fda68aff6
Fix repo id bug
...
This to fix this issue #1482
2022-10-05 16:26:06 +02:00
b8c8550841
Data improvements. Improve train support for in_chans != 3. Add wds dataset support from bits_and_tpu branch w/ fixes and tweaks. TFDS tweaks.
2022-09-29 16:42:58 -07:00
7327792f39
update to support pickle based dictionaries
2022-09-27 11:13:48 -04:00
1199c5a1a4
clip_laion2b models need 1e-5 eps for LayerNorm
2022-09-25 10:36:54 -07:00
87939e6fab
Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.
2022-09-23 16:08:59 -07:00
c88947ad3d
Add initial Hugging Face Datasets parser impl.
2022-09-23 16:08:19 -07:00
e858912e0c
Add brute-force checkpoint remapping option
2022-09-23 16:07:03 -07:00
b293dfa595
Add CL SE module
2022-09-23 16:06:09 -07:00
2a296412be
Add Adan optimizer
2022-09-23 16:05:52 -07:00
5dc4343308
version 0.6.11
2022-09-23 13:54:56 -07:00
a383ef99f5
Make huggingface_hub necessary if it's the only source for a pretrained weight
2022-09-23 13:54:21 -07:00
33e30f8c8b
Remove layer-decay print
2022-09-18 21:33:03 -07:00
e069249a2d
Add hf hub entries for laion2b clip models, add huggingface_hub dependency, update some setup/reqs, torch >= 1.7
2022-09-16 21:39:05 -07:00
9d65557be3
Fix errant import
2022-09-15 17:47:23 -07:00
9709dbaaa9
Adding support for fine-tune CLIP LAION-2B image tower weights for B/32, L/14, H/14 and g/14. Still WIP
2022-09-15 17:25:59 -07:00
a520da9b49
Update tresnet features_info for v2
2022-09-13 20:54:54 -07:00
c8ab747bf4
BEiT-V2 checkpoints didn't remove 'module' from weights, adapt checkpoint filter
2022-09-13 17:56:49 -07:00
73049dc2aa
Fix type in dla weight update
2022-09-13 17:52:45 -07:00
3599c7e6a4
version 0.6.10
2022-09-13 16:37:02 -07:00
e11efa872d
Update a bunch of weights with external links to timm release assets. Fixes issue with *aliyuncs.com returning forbidden. Did pickle scan / verify and re-hash. Add TresNet-V2-L weights.
2022-09-13 16:35:26 -07:00
fa8c84eede
Update maxvit_tiny_256 weight to better iter, add coatnet / maxvit / maxxvit model defs for future runs
2022-09-07 12:37:37 -07:00
c1b3cea19d
Add maxvit_rmlp_tiny_rw_256 model def and weights w/ 84.2 top-1 @ 256, 84.8 @ 320
2022-09-07 10:27:11 -07:00
914544fc81
Add beitv2 224x224 checkpoints from https://github.com/microsoft/unilm/tree/master/beit2
2022-09-06 20:25:18 -07:00
dc90816f26
Add `maxvit_tiny_rw_224` weights 83.5 @ 224 and `maxvit_rmlp_pico_rw_256` relpos weights, 80.5 @ 256, 81.3 @ 320
2022-09-06 16:14:41 -07:00
f489f02ad1
Make gcvit window size ratio based to improve resolution changing support #1449 . Change default init to original.
2022-09-06 16:14:00 -07:00
7f1b223c02
Add maxvit_rmlp_nano_rw_256 model def & weights, make window/grid size dynamic wrt img_size by default
2022-08-29 15:49:32 -07:00
e6a4361306
pretrained_cfg entry for mvitv2_small_cls
2022-08-28 15:27:01 -07:00
f66e5f0e35
Fix class token support in MViT-V2, add small_class variant to ensure it's tested. Fix #1443
2022-08-28 15:24:04 -07:00
f1d2160d85
Update a few maxxvit comments, rename PartitionAttention -> PartitionAttenionCl for consistency with other blocks
2022-08-26 12:53:49 -07:00
eca6f0a25c
Fix syntax error (extra dataclass comma) in maxxvit.py
2022-08-26 11:29:09 -07:00
ff6a919cf5
Add --fast-norm arg to benchmark.py, train.py, validate.py
2022-08-25 17:20:46 -07:00
769ab4b98a
Clean up no_grad for trunc normal weight inits
2022-08-25 16:29:52 -07:00
48e1df8b37
Add norm/norm_act header comments
2022-08-25 16:29:34 -07:00
7c2660576d
Tweak init for convnext block using maxxvit/coatnext.
2022-08-25 15:30:59 -07:00
1d8d6f6072
Fix two default args in DenseNet blocks... fix #1427
2022-08-25 15:00:35 -07:00
527f9a4cb2
Updated to correct maxvit_nano weights...
2022-08-24 12:42:11 -07:00
b2e8426fca
Make k=stride=2 ('avg2') pooling default for coatnet/maxvit. Add weight links. Rename 'combined' partition to 'parallel'.
2022-08-24 11:01:20 -07:00
837c68263b
For ConvNeXt, use timm internal LayerNorm for fast_norm in non conv_mlp mode
2022-08-23 15:17:12 -07:00
cac0a4570a
More test fixes, pool size for 256x256 maxvit models
2022-08-23 13:38:26 -07:00
e939ed19b9
Rename internal creation fn for maxvit, has not been just coatnet for a while...
2022-08-22 17:44:51 -07:00
ffaf97f813
MaxxVit! A very configurable MaxVit and CoAtNet impl with lots of goodies..
2022-08-22 17:42:10 -07:00
8c9696c9df
More model and test fixes
2022-08-22 17:40:31 -07:00
ca52108c2b
Fix some model support functions
2022-08-19 10:20:51 -07:00
f332fc2db7
Fix some test failures, torchscript issues
2022-08-18 16:19:46 -07:00
6e559e9b5f
Add MViT (Multi-Scale) V2
2022-08-17 15:12:31 -07:00
43aa84e861
Add 'fast' layer norm that doesn't cast to float32, support APEX LN impl for slight speed gain, update norm and act factories, tweak SE for ability to disable bias (needed by GCVit)
2022-08-17 14:32:58 -07:00
c486aa71f8
Add GCViT
2022-08-17 14:29:18 -07:00
fba6ecd39b
Add EfficientFormer
2022-08-17 14:08:53 -07:00
ff4a38e2c3
Add PyramidVisionTransformerV2
2022-08-17 12:06:05 -07:00
1d8ada359a
Add timm ConvNeXt 'atto' weights, change test resolution for FB ConvNeXt 224x224 weights, add support for different dw kernel_size
2022-08-15 17:56:08 -07:00
2544d3b80f
ConvNeXt pico, femto, and nano, pico, femto ols (overlapping stem) weights and model defs
2022-08-05 17:05:50 -07:00
13565aad50
Add edgenext_base model def & weight link, update to improve ONNX export #1385
2022-08-05 16:58:34 -07:00
8ad4bdfa06
Allow ntuple to be used with string values
2022-07-28 16:18:18 -07:00
faae93e62d
Fix typo in PositionalEncodingFourier
2022-07-28 19:08:08 -04:00
7430a85d07
Update README, bump version to 0.6.8
2022-07-28 15:07:11 -07:00
ec6a28830f
Add DeiT-III 'medium' model defs and weights
2022-07-28 15:03:20 -07:00
d875a1d3f6
version 0.6.7
2022-07-27 12:41:06 -07:00
6f103a442b
Add convnext_nano weights, 80.8 @ 224, 81.5 @ 288
2022-07-26 16:40:27 -07:00
4042a94f8f
Add weights for two 'Edge' block (3x3->1x1) variants of CS3 networks.
2022-07-26 16:40:27 -07:00
c8f69e04a9
Merge pull request #1365 from veritable-tech/fix-resize-pos-embed
...
Take `no_emb_class` into account when calling `resize_pos_embed`
2022-07-24 21:03:01 -07:00
0b64117592
Take `no_emb_class` into account when calling `resize_pos_embed`
2022-07-24 19:11:45 +08:00
Ross Wightman
hongxin xiang
Wauplin
Hoan Nguyen
Ross Wightman
nateraw
Mohamed Rashad
Alex Fafard
Christoph Reich
Ceshine Lee