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,711 Commits
136 Branches
65 Tags
47 MiB
Commit Graph

2711 Commits (c8c4f256b8c279575ed87012f38620a4cd25df57)
 

Author SHA1 Message Date
Ross Wightman bd0f79153b Add weights only flag to avoid warning, try to keep bwd compat. Default to True for remote load of pretrained weights, keep False for local checkpoing load to avoid training checkpoint breaks.. fix #2249 2024-08-12 11:41:41 -07:00
Ross Wightman 531215eded Bring changelog up to current, back to dev version, 1.0.9.dev0 2024-08-08 09:17:23 -07:00
Ross Wightman d29b720d8d Add changes to hfdocs toctree 2024-08-08 09:12:58 -07:00
Ross Wightman 4fb878bdd6 Remove old docs 2024-08-08 09:08:56 -07:00
Ross Wightman f96e369615
Update changes.md 2024-08-08 08:37:49 -07:00
Ross Wightman 33ab2b35ae
Update README.md 2024-08-08 08:36:04 -07:00
Ross Wightman bf781bc3c2
Merge pull request #2245 from dhkim0225/feature/rdnet 
Feature: Add RDNet
 2024-08-07 16:19:22 -07:00
Ross Wightman 49f6e1da96 Remove eval to avoid safety concerns 2024-08-07 14:54:13 -07:00
Ross Wightman 2d5c9bf60d
Merge pull request #2253 from brendanartley/hgnet-grad-checkpointing 
Add gradient checkpointing to hgnets
 2024-08-07 12:45:14 -07:00
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 3b6a1d4d48 model summary added 2024-08-02 09:11:44 +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 a6fe31b096 Update README.md 2024-07-28 22:16:06 -07: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 9558a7f8eb
Update README.md 2024-07-26 10:16:44 -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 8b14fc7bb6
Merge pull request #2240 from Zirunis/Zirunis-patch-1 
Fix LR scheduler help in train.py
 2024-07-23 11:04:18 -07:00
Ross Wightman f3c11dc3a5
Merge pull request #2239 from huggingface/fix_out_indices_order 
Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet
 2024-07-23 11:02:42 -07:00
Ross Wightman 8efdc38213 Fix #2242 add checks for out indices with intermediate getter mode 2024-07-23 08:19:09 -07:00
Zirunis 4ed93fce93
Fix LR scheduler help in train.py 
The default is and always has been the cosine scheduler, yet the help states that the default would be the step scheduler. Whatever the intended one was, for backwards compatibility the default should definitely remain cosine, which is why I changed the help comment to reflect that.
 2024-07-22 23:04:00 +02: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 a1996ec0f4
Merge pull request #2238 from huggingface/fix_mnv4_query_strides 
Fix mnv4 query strides
 2024-07-19 16:32:08 -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
Ross Wightman 474c9cf768
Merge pull request #2236 from NightMachinery/patch-1 
eva.py: fixed bug in applying attention mask
 2024-07-19 08:09:56 -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
Ross Wightman 3196d6b131
Merge pull request #2230 from TianyiFranklinWang/main 
Avoid zero division error
 2024-07-14 21:35:24 -07:00
Tianyi Wang d3ce5a8665
Avoid zero division error 2024-07-15 12:45:46 +10:00
Promisery 417cf7f871 Initialize weights of reg_token for ViT 2024-07-13 11:11:42 +08:00