mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
2,681 Commits
136 Branches
65 Tags
47 MiB
Commit Graph

2681 Commits (9ab5464e4d7fa6bd2350eda7d93d5ec3fe8b9927)
 

Author SHA1 Message Date
Ross Wightman 9ab5464e4d More additions to Kron 2025-01-27 21:02:26 -08:00
Ross Wightman 5f10450235 Some more kron work. Figured out why some tests fail, implemented a deterministic rng state load but too slow so skipping some tests for now. 2025-01-27 21:02:26 -08:00
Ross Wightman cd21e80d03 Fiddling with Kron (PSGD) 2025-01-27 21:02:26 -08:00
Adam J. Stewart d81da93c16 Use import alias 2025-01-22 10:27:17 -08:00
Adam J. Stewart 4de1abf837 timm: add __all__ to __init__ 2025-01-22 10:27:17 -08:00
Ryan bda46f8e6f Add num_classes assertion after reset_classifier 2025-01-21 11:52:05 -08:00
Ryan 17eabaad17 Fix RDNet forward call 2025-01-21 11:52:05 -08:00
Ryan 80a4877376 Fix self.reset_classifier num_classes update 2025-01-21 11:52:05 -08:00
Collin McCarthy 84631cb5c6 Add missing training flag to convert_sync_batchnorm 2025-01-21 11:51:55 -08:00
Josua Rieder cb4cea561a add arguments to the respective argument groups 2025-01-20 10:54:35 -08:00
Josua Rieder 634b68ae50 Fix metavar for --input-size 2025-01-20 10:53:46 -08:00
Ross Wightman 5d535d7a2d Version 1.0.14, update README & changelog 2025-01-19 13:53:09 -08:00
Ross Wightman c6b74eb5bd Remove numpy ver constraint, not relevant for latest PyTorch version 2025-01-19 13:41:16 -08:00
Ross Wightman aa333079da Tweak so150m2 def 2025-01-19 13:40:53 -08:00
Josua Rieder 8d81fdf3d9 Fix typos 2025-01-19 13:39:40 -08:00
Ross Wightman 3677f67902 Add the 256x256 in1k ft of the so150m, add an alternate so150m def 2025-01-18 15:51:57 -08:00
Ross Wightman 2a84d68d02 Add some so150m vit w/ sbb recipe weights, and a ese_vovnet57b model with RA4 recipe 2025-01-18 15:51:57 -08:00
Ross Wightman 9265d54a3a LeViT safetensors load is broken by conversion code that wasn't deactivated 2025-01-16 11:37:00 -08:00
Ross Wightman 21e75a9d25
Update version.py 
Back to dev version
 2025-01-16 11:23:17 -08:00
Ross Wightman c96e9e7ce0
Merge pull request #2408 from JosuaRieder/no_console_results 
Implement --no-console-results in inference.py
 2025-01-15 08:11:31 -08:00
Ross Wightman 63b2de7a6c
Merge pull request #2409 from adamjstewart/models/vgg 
VGG ConvMlp: fix layer defaults/types
 2025-01-15 07:37:26 -08:00
Adam J. Stewart 6d21eb0d37
VGG ConvMlp: fix layer defaults/types 2025-01-15 12:11:56 +01:00
Josua Rieder 47790637f2 Implement --no-console-results in inference.py 2025-01-15 11:37:11 +01:00
Ross Wightman ef7dec8e43
Merge pull request #2406 from JosuaRieder/fix_latex 
fix incorrect LaTeX formulas
 2025-01-14 11:03:15 -08:00
Ross Wightman 1572769059
Merge pull request #2402 from JosuaRieder/fix_inference_csv_export 
disable abbreviating csv inference output with ellipses
 2025-01-14 11:01:48 -08:00
Ross Wightman fc0609bcb6 Add --model-dtype (pure bfloat16/float16) support to inference.py 2025-01-14 11:00:16 -08:00
Ross Wightman 53c3c89a86
Merge pull request #2403 from JosuaRieder/efficientnet_typo 
fix typo in EfficientNet docs
 2025-01-14 10:57:46 -08:00
Ross Wightman 5e0df16fc0
Merge pull request #2404 from JosuaRieder/fix_training_scripts_doc_link 
fix 'timm recipe scripts' link
 2025-01-14 10:57:25 -08:00
Ross Wightman 7fa37c1f26
Merge pull request #2405 from JosuaRieder/add_missing_paper_title 
Add missing paper title
 2025-01-14 10:57:02 -08:00
Josua Rieder adf9efeac7 fix incorrect LaTeX formulas 2025-01-14 19:17:52 +01:00
Josua Rieder 1dab96c637 fix small copy paste mistake 2025-01-14 18:58:54 +01:00
Josua Rieder cac1899ba1 add missing paper title DLA: Deep Layer Aggregation 2025-01-14 18:56:23 +01:00
Josua Rieder 676c734796 fix 'timm recipe scripts' link 2025-01-14 18:49:45 +01:00
Josua Rieder 17a1abfc0d fix typo in EfficientNet docs 2025-01-14 18:36:51 +01:00
Josua Rieder 8ce197e33a disable abbreviating csv inference output with ellipses 2025-01-14 17:39:48 +01:00
Ross Wightman ff77dfa825
Merge pull request #2400 from adamjstewart/types/nn-module 
Fix nn.Module type hints
 2025-01-14 08:23:57 -08:00
Adam J. Stewart f5c4d5cbb7
Add missing imports 2025-01-11 15:13:16 +01:00
Adam J. Stewart 19aaea3c8f
Fix nn.Module type hints 2025-01-11 15:09:21 +01:00
Ross Wightman 47811bc05a Update README, bump version to 1.0.13 non-dev 2025-01-09 09:33:59 -08:00
Ross Wightman eeee38e972 Avoid unecessary compat break btw train script and nearby timm versions w/ dtype addition. 2025-01-08 21:10:15 -08:00
Ross Wightman deb9895600 Update checkpoint save to fix old hard-link + fuse issue I ran into again... fix #340 2025-01-08 15:36:58 -08:00
Ross Wightman c4fb98f399
Merge pull request #2398 from huggingface/caojiaolong-main 
Merging wandb project name chages w/ addition
 2025-01-08 10:15:09 -08:00
Ross Wightman c173886e75 Merge branch 'main' into caojiaolong-main 2025-01-08 09:11:50 -08:00
Ross Wightman 2d0ac6f567
Merge pull request #2397 from huggingface/half_prec_trainval 
Add half-precision (bfloat16, float16) support to train & validate scripts
 2025-01-07 11:48:02 -08:00
Ross Wightman 1969528296 Fix dtype log when default (None) is used w/o AMP 2025-01-07 11:47:22 -08:00
Ross Wightman 92f610c982 Add half-precision (bfloat16, float16) support to train & validate scripts. Should push dtype handling into model factory / pretrained load at some point... 2025-01-07 10:25:14 -08:00
Jiao-Long Cao 40c19f3939
Add wandb project name argument and allow change wandb run name 2025-01-07 16:43:34 +08:00
Ross Wightman 6f80214e80
Merge pull request #2394 from huggingface/non_reentrant_ckpt 
Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override
 2025-01-06 14:44:06 -08:00
Ross Wightman 155f6e7fea Update README, few minor fixups. 2025-01-06 13:09:15 -08:00
Ross Wightman 2b251fb291 Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override 2025-01-06 11:28:39 -08:00