1670 Commits (01fdf444386fe63a5be9ad3767457bb3a6c2b707)

 

All Branches   

Search

Author	SHA1	Message	Date
Ross Wightman	2f2b22d8c7	Disable nvfuser fma / opt level overrides per #1244	3 years ago
Ross Wightman	c0211b0bf7	Swin-V2 test fixes, typo	3 years ago
Ross Wightman	9a86b900fa	Official SwinV2 models	3 years ago
Ross Wightman	d07d015173	Merge pull request #1249 from okojoalg/sequencer ... Add Sequencer	3 years ago
Ross Wightman	d30685c283	Merge pull request #1251 from hankyul2/fix-multistep-scheduler ... fix: multistep lr decay epoch bugs	3 years ago
han	a16171335b	fix: change milestones to decay-milestones ... - change argparser option `milestone` to `decay-milestone`	3 years ago
Ross Wightman	39b725e1c9	Fix tests for rank-4 output where feature channels dim is -1 (3) and not 1	3 years ago
Ross Wightman	78a32655fa	Fix poolformer group_matcher to merge proj downsample with previous block, support coarse	3 years ago
Ross Wightman	d79f3d9d1e	Fix torchscript use for sequencer, add group_matcher, forward_head support, minor formatting	3 years ago
Ross Wightman	37b6920df3	Fix group_matcher regex for regnet.py	3 years ago
okojoalg	93a79a3dd9	Fix num_features in Sequencer	3 years ago
okojoalg	2fec08e923	Add Sequencer to non std filters	3 years ago
han	57a988df30	fix: multistep lr decay epoch bugs ... - add milestones arguments - change decay_epochs to milestones variable	3 years ago
okojoalg	578d52e752	Add Sequencer	3 years ago
Ross Wightman	6d4665bb52	Merge pull request #1245 from rwightman/vit_relpos_refactor ... Vision Transformer refactoring and Rel Pos impl	3 years ago
Ross Wightman	f5ca4141f7	Adjust arg order for recent vit model args, add a few comments	3 years ago
Ross Wightman	41dc49a337	Vision Transformer refactoring and Rel Pos impl	3 years ago
Ross Wightman	b7cb8d0337	Add Swin-V2 Small-NS weights (83.5 @ 224). Add layer scale like 'init_values' via post-norm LN weight scaling	3 years ago
Ross Wightman	001688dabf	Merge pull request #1233 from jjsjann123/nhwc_cond_conv2d ... fixing channels_last on cond_conv2d; update nvfuser debug env variable	3 years ago
jjsjann123	f88c606fcf	fixing channels_last on cond_conv2d; update nvfuser debug env variable	3 years ago
Ross Wightman	7d235c5a5f	Merge pull request #1230 from donglixp/patch-1 ... migrate azure blob for beit checkpoints	3 years ago
Li Dong	09e9f3defb	migrate azure blob for beit checkpoints ... ## Motivation We are going to use a new blob account to store the checkpoints. ## Modification Modify the azure blob storage URLs for BEiT checkpoints.	3 years ago
Ross Wightman	52ac881402	Missed first_conv in latest seresnext 'D' default_cfgs	3 years ago
Ross Wightman	7629d8264d	Add two new SE-ResNeXt101-D 32x8d weights, one anti-aliased and one not. Reshuffle default_cfgs vs model entrypoints for resnet.py so they are better aligned.	3 years ago
Ross Wightman	fbf597049c	Update README and change timmdocs link in documentation	3 years ago
Ross Wightman	01a0e25a67	Merge pull request #1208 from seefun/master	3 years ago
SeeFun	8f0bc0591e	fix convnext args	3 years ago
SeeFun	b0d2fcf647	Merge branch 'rwightman:master' into master	3 years ago
Ross Wightman	eac2df3d2c	Update PyTorch 1.10 benchmark numbers for latest code	3 years ago
Ross Wightman	c5a8e929fb	Add initial swinv2 tiny / small weights	3 years ago
Ross Wightman	83d7a11eec	Update .gitignore, remove out of date notebooks	3 years ago
Ross Wightman	02b806e00a	Add PyTorch 1.11 train benchmark numbers	3 years ago
Ross Wightman	c9f208f7f8	Update eval results and add latest PyTorch 1.11 inference benchmarks	3 years ago
Ross Wightman	f670d98cb8	Make a few more layers symbolically traceable (remove from FX leaf modules) ... * remove dtype kwarg from .to() calls in EvoNorm as it messed up script + trace combo * BatchNormAct2d always uses custom forward (cut & paste from original) instead of super().forward. Fixes #1176 * BlurPool groups==channels, no need to use input.dim[1]	3 years ago
Ross Wightman	a9ecb880e5	Merge pull request #1190 from seefun/ConvNeXt-pretrain ... Add ConvNeXt tiny and small pretrain in22k	3 years ago
SeeFun	5f4de2334b	Merge pull request #1 from seefun/ConvNeXt-pretrain ... Add ConvNeXt tiny and small pretrain in22k	3 years ago
SeeFun	ec4e9aa5a0	Add ConvNeXt tiny and small pretrain in22k ... Add ConvNeXt tiny and small pretrain in22k from ConvNeXt repo: 06f7b05f92	3 years ago
Ross Wightman	73ffade1f8	Update README.md	3 years ago
Ross Wightman	575924ed60	Update test crop for new RegNet-V weights to match Y	3 years ago
Ross Wightman	8f6d638887	Update README.md	3 years ago
Ross Wightman	1618527098	Add layer scale and parallel blocks to vision_transformer	3 years ago
Ross Wightman	c42be74621	Add attrib / comments about Swin-S3 (AutoFormerV2) weights	3 years ago
Ross Wightman	474ac906a2	Add 'head norm first' convnext_tiny_hnf weights	3 years ago
Ross Wightman	dc51334cdc	Fix pruned adapt for EfficientNet models that are now using BatchNormAct layers	3 years ago
Ross Wightman	024fc4d9ab	version 0.6.1 for master	3 years ago
Ross Wightman	e1e037ba52	Fix bad tuple typing fix that was on XLA branch bust missed on master merge	3 years ago
Ross Wightman	341b464a5a	Remove redundant noise attr from Plateau scheduler (use parent)	3 years ago
Ross Wightman	7514439573	Merge pull request #1014 from rwightman/norm_norm_norm ... Normalization layer additions, model API updates, new models, new weights, and enhancements	3 years ago
Ross Wightman	ff21fdb41d	Update README.md ready for merge	3 years ago
Ross Wightman	fe457c1996	Update SwinTransformerV2Cr post-merge, update with grad checkpointing / grad matcher ... * weight compat break, activate norm3 for final block of final stage (equivalent to pre-head norm, but while still in BLC shape) * remove fold/unfold for TPU compat, add commented out roll code for TPU * add option for end of stage norm in all stages * allow weight_init to be selected between pytorch default inits and xavier / moco style vit variant	3 years ago