922 Commits (dff33730b38aa8e923da14fef34f7c84c9431c18)

Author	SHA1	Message	Date
Ross Wightman	dff33730b3	Merge remote-tracking branch 'origin/master' into bits_and_tpu	3 years ago
Ross Wightman	20a1fa63f8	Make dev version 0.6.2.dev0 for pypi pre	3 years ago
Ross Wightman	347308faad	Update README.md, version to 0.6.2	3 years ago
Ross Wightman	4b30bae67b	Add updated vit_relpos weights, and impl w/ support for official swin-v2 differences for relpos. Add bias control support for MLP layers	3 years ago
Ross Wightman	d4c0588012	Remove persistent buffers from Swin-V2. Change SwinV2Cr cos attn + tau/logit_scale to match official, add ckpt convert, init_value zeros resid LN weight by default	3 years ago
Ross Wightman	27c42f0830	Fix torchscript use for offician Swin-V2, add support for non-square window/shift to WindowAttn/Block	3 years ago
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
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	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	9c321be330	Merge pull request #1239 from dedeswim/parser-fix ... Add support for different TFDS `BuilderConfig`s	3 years ago
Edoardo Debenedetti	c76d772670	Add support for different TFDS `BuilderConfig`s	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
jjsjann123	f88c606fcf	fixing channels_last on cond_conv2d; update nvfuser debug env variable	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
SeeFun	8f0bc0591e	fix convnext args	3 years ago
Ross Wightman	754e11402a	Merge remote-tracking branch 'origin/master' into bits_and_tpu	3 years ago
Ross Wightman	c5a8e929fb	Add initial swinv2 tiny / small weights	3 years ago
Ross Wightman	1ba0ec4c18	Merge remote-tracking branch 'origin/master' into bits_and_tpu	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
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	575924ed60	Update test crop for new RegNet-V weights to match Y	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	749856cf25	Merge remote-tracking branch 'origin/norm_norm_norm' into bits_and_tpu	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
Ross Wightman	b049a5c5c6	Merge remote-tracking branch 'origin/master' into norm_norm_norm	3 years ago
Ross Wightman	7cdd164d77	Fix #1184, scheduler noise bug during merge madness	3 years ago
Ross Wightman	9440a50c95	Merge branch 'mrT23-master'	3 years ago
Ross Wightman	d98aa47d12	Revert ml-decoder changes to model factory and train script	3 years ago
Ross Wightman	b20665d379	Merge pull request #1007 from qwertyforce/patch-1 ... update arxiv link	3 years ago
Ross Wightman	7a0994f581	Merge pull request #1150 from ChristophReich1996/master ... Swin Transformer V2	3 years ago