mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
2,118 Commits 141 Branches 65 Tags
Commit Graph

2118 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Ross Wightman
c50004db79 Allow training w/o validation split set 2024-01-08 09:38:42 -08:00
Ross Wightman
be0944edae Significant transforms, dataset, dataloading enhancements. 2024-01-08 09:38:42 -08:00
Ross Wightman
b5a4fa9c3b Add pos_weight and support for summing over classes to BCE impl in train scripts 2023-12-30 12:13:06 -08:00
Ross Wightman
f2fdd97e9f Add parsable json results output for train.py, tweak --pretrained-path to force head adaptation 2023-12-22 11:18:25 -08:00
LR
e0079c92da
Update eva.py (#2058) 
* Update eva.py

When argument class token = False, self.cls_token = None.

Prevents error from attempting trunc_normal_ on None:
AttributeError: 'NoneType' object has no attribute 'uniform_'

* Update eva.py

fix
 2023-12-16 15:10:45 -08:00
Li zhuoqun
7da34a999a add type annotations in the code of swin_transformer_v2 2023-12-15 09:31:25 -08:00
Fredo Guan
bbe798317f
Update EdgeNeXt to use ClassifierHead as per ConvNeXt (#2051) 
* Update edgenext.py
 2023-12-11 12:17:19 -08:00
Ross Wightman
711c5dee6d Update sgdw for older pytorch 2023-12-11 12:10:29 -08:00
Ross Wightman
60b170b200 Add --pretrained-path arg to train script to allow passing local checkpoint as pretrained. Add missing/unexpected keys log. 2023-12-11 12:10:29 -08:00
Ross Wightman
17a47c0e35 Add SGDW optimizer 2023-12-11 12:10:29 -08:00
Fredo Guan
2597ce2860 Update davit.py 2023-12-11 11:13:04 -08:00
Ross Wightman
c82598fc5e Remove deprecated doc delete workflows 2023-12-05 12:04:38 -08:00
akiyuki ishikawa
2bd043ce5d fix doc position 2023-12-05 12:00:51 -08:00
akiyuki ishikawa
4f2e1bf4cb Add missing docs in SwinTransformerStage 2023-12-05 12:00:51 -08:00
Ross Wightman
df7ae11eb2 Add device arg for patch embed resize, fix #2024 2023-12-04 11:42:13 -08:00
Ross Wightman
cd8d9d9ff3 Add missing hf hub entries for mvitv2 2023-11-26 21:06:39 -08:00
Ross Wightman
19a8c182cc Version 0.9.13dev0 2023-11-25 10:52:31 -08:00
Ross Wightman
79548b542f Update PT 2.1 inference benchmarks w/ full profile info 2023-11-25 10:52:03 -08:00
Ross Wightman
b996c1a0f5 A few more missed hf hub entries v0.9.12 2023-11-23 21:48:14 -08:00
Ross Wightman
f9c9f6b7d8 Update README 2023-11-23 17:10:27 -08:00
Ross Wightman
5fb92d79e7 Version 0.9.12 2023-11-23 17:01:03 -08:00
Ross Wightman
89ec91aece Add missing hf_hub entry for mobilnetv3_rw 2023-11-23 12:44:59 -08:00
Ross Wightman
e7abc320f5 Improve bulk_runner for 'all resolution' mode 2023-11-23 12:44:59 -08:00
Ross Wightman
40d55ab4bc Add in_chans to data config helper. Fix #2021 2023-11-23 12:44:59 -08:00
Dillon Laird
63ee54853c fixed intermediate output indices 2023-11-22 16:32:41 -08:00
Ross Wightman
4d0737d5fa Add PT 2.1 NCHW 3090 benchmark results 2023-11-21 16:49:20 -08:00
Ross Wightman
fa06f6c481 Merge branch 'seefun-efficientvit' 2023-11-21 14:06:27 -08:00
Ross Wightman
c6b0c98963 Upload weights to hub, tweak crop_pct, comment out SAM EfficientViTs for now (v2 weights comming) 2023-11-21 14:05:04 -08:00
Ross Wightman
975203a369 Version 0.9.12dev0 2023-11-21 10:18:45 -08:00
Ross Wightman
055522bf84 Going to stop supporting Python 3.7 soon 2023-11-21 09:48:03 -08:00
Ross Wightman
ada145b016 Literal use w/ python < 3.8 requires typing_extension, cach instead of check sys ver 2023-11-21 09:48:03 -08:00
Ross Wightman
dfaab97d20 More consistency in model arg/kwarg merge handling 2023-11-21 09:48:03 -08:00
Ross Wightman
3775e4984f Merge branch 'efficientvit' of github.com:seefun/pytorch-image-models into seefun-efficientvit 2023-11-20 16:21:38 -08:00
Ross Wightman
3370053e71 Update README v0.9.11 2023-11-20 14:39:22 -08:00
Ross Wightman
25cf2c2dbb Update bulk_runner with improved filtering options for benchmarking / val runs 2023-11-20 14:08:51 -08:00
Ross Wightman
dfb8658100 Fix a few missed model deprecations and one missed pretrained cfg 2023-11-20 12:41:49 -08:00
Ross Wightman
057fdb570b Update model eval result csv files 2023-11-20 10:06:40 -08:00
Ross Wightman
c20f5fc385 Version 0.9.11 2023-11-19 17:18:48 -08:00
Ross Wightman
a604011935 Add support for passing model args via hf hub config 2023-11-19 15:16:01 -08:00
Laureηt
23e7f17724 replace inline latex syntax in hfdocs 2023-11-19 09:34:38 -08:00
方曦
c9d093a58e update norm eps for efficientvit large 2023-11-18 17:46:47 +08:00
Laureηt
21647c0a0c Add types to vision_transformers.py 2023-11-17 16:06:06 -08:00
方曦
87ba43a9bc add efficientvit large series 2023-11-17 13:58:46 +08:00
Alex Cunha
a38c2b4475 fix bug 2023-11-16 20:27:59 -08:00
Alex Cunha
bbf0ab000c return if cuda not available 2023-11-16 20:27:59 -08:00
Alex Cunha
e628ed7e67 device agnostic testing 2023-11-16 20:27:59 -08:00
Ross Wightman
7c685a4ef3 Fix openai quickgelu loading and add mnissing orig_in21k vit weights and remove zero'd classifier w/ matching hub update 2023-11-16 19:16:28 -08:00
LittleNyima
ef72c3cd47 Add warnings for duplicate registry names 2023-11-08 10:18:59 -08:00
Ross Wightman
0bc9a63ee5
Update README.md 2023-11-07 11:28:23 -08:00
Ross Wightman
205d8ad37c version 0.9.10 v0.9.10 2023-11-04 02:33:04 -07:00