1281 Commits (5f81d4de234f579bdc988e8346da14b37a3af160)

 

All Branches   

Search

Author	SHA1	Message	Date
Ross Wightman	ecc7552c5c	Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup.	4 years ago
Ross Wightman	165fb354b2	Add initial RedNet model / Involution layer impl for testing	4 years ago
Ross Wightman	328249f11a	Update README, tweak fine-tune effv2 model names.	4 years ago
Ross Wightman	c4f482a08b	EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models.	4 years ago
Ross Wightman	c16d65a8a7	Update results.csv files with cait, mlp-mixer, resnetrs, miil imagenet21k pretrained, and my latest efficientnet v2s/b4 weights.	4 years ago
Ross Wightman	4fbc32d3d0	Fix crop_pct for cait models.	4 years ago
comar	d7e1e7144a	fix: the exeption when topk is 1	4 years ago
comar	ffa90e04d3	fix: the exception not using default topk argument	4 years ago
Ross Wightman	715519a5ef	Rethink name of patch embed grid info	4 years ago
Ross Wightman	b2c305c2aa	Move Mlp and PatchEmbed modules into layers. Being used in lots of models now...	4 years ago
Ross Wightman	3ba6b55cb2	More adjustments to ByoaNet models for further experiments.	4 years ago
Ross Wightman	f480734ad5	Update README.md	4 years ago
Ross Wightman	504fef4bc1	Merge pull request #609 from rwightman/cait ... CaiT transformer model	4 years ago
Ross Wightman	d45e50b9db	Update test for cait 448x448 model	4 years ago
Ross Wightman	5fcddb96a8	Merge branch 'master' into cait	4 years ago
Ross Wightman	3db12b4b6a	Finish CaiT cleanup	4 years ago
Ross Wightman	936e9b331d	Merge pull request #613 from rwightman/mlp-mixer ... MLP-Mixer models w/ pretrained weights ported from Google JAX impl...	4 years ago
Ross Wightman	2d8b09fe8b	Add official pretrained weights to MLP-Mixer, complete model cfgs.	4 years ago
Ross Wightman	12efffa6b1	Initial MLP-Mixer attempt...	4 years ago
Ross Wightman	0721559511	Improved (hopefully) init for SA/SA-like layers used in ByoaNets	4 years ago
Ross Wightman	d5473c17f7	Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.	4 years ago
Ross Wightman	0d87650fea	Remove filter hack from BlurPool w/ non-persistent buffer. Use BlurPool2d instead of AntiAliasing.. for TResNet. Breaks PyTorch < 1.6.	4 years ago
Ross Wightman	ddc743fdf8	Update ResNet-RS models to EMA weights	4 years ago
Ross Wightman	08d60f4a9a	resnetrs50 pool sizing wrong	4 years ago
Ross Wightman	137a374930	Merge pull request #555 from MichaelMonashev/patch-1 ... benchmark.py argument description fixed	4 years ago
Ross Wightman	1daa15ecc3	Initial Cait commit. Still some cleanup to do.	4 years ago
Ross Wightman	67d0665b46	Post ResNet-RS merge cleanup. Add weight urls, adjust train/test/crop pct.	4 years ago
Aman Arora	560eae38f5	[WIP] Add ResNet-RS models (#554) ... * Add ResNet-RS models * Only include resnet-rs changes * remove whitespace diff * EOF newline * Update time * increase time * Add first conv * Try running only resnetv2_101x1_bitm on Linux runner * Add to exclude filter * Run test_model_forward_features for all * Add to exclude ftrs * back to defaults * only run test_forward_features * run all tests * Run all tests * Add bigger resnetrs to model filters to fix Github CLI * Remove resnetv2_101x1_bitm from exclude feat features * Remove hardcoded values * Make sure reduction ratio in resnetrs is 0.25 * There is no bias in replaced maxpool so remove it	4 years ago
Ross Wightman	9cc7dda6e5	Fixup byoanet configs to pass unit tests. Add swin_attn and swinnet26t model for testing.	4 years ago
Ross Wightman	e15c3886ba	Defaul lambda r=7. Define '26t' stage 4/5 256x256 variants for all of bot/halo/lambda nets for experiment. Add resnet50t for exp. Fix a few comments.	4 years ago
Ross Wightman	d15ad3e919	Merge branch 'morizin-master'	4 years ago
Ross Wightman	e5e15754c9	Fix coat first conv ident	4 years ago
Ross Wightman	76739a7589	CoaT merge. Bit of formatting, fix torchscript (for non features), remove einops/einsum dep, add pretrained weight hub (url) support.	4 years ago
Ross Wightman	026430c083	Merge branch 'master' of https://github.com/morizin/pytorch-image-models-1 into morizin-master	4 years ago
Ross Wightman	c03e45cbf3	Merge branch 'mrT23-master'	4 years ago
Ross Wightman	a0492e3b48	A few miil weights naming tweaks to improve compat with model registry and filtering wildcards.	4 years ago
talrid	8c1f03e56c	comment	4 years ago
talrid	19e1b67a84	old spaces	4 years ago
talrid	a443865876	update naming and scores	4 years ago
talrid	cf0e371594	84_0	4 years ago
talrid	0968bdeca3	vit, tresnet and mobilenetV3 ImageNet-21K-P weights	4 years ago
mrT23	b81cd75eea	Merge pull request #3 from rwightman/master ... merge base	4 years ago
morizin	1e3b6d4dfc	Update __init__.py	4 years ago
morizin	fd022fd6a2	Update __init__.py	4 years ago
morizin	06841427cd	Add files via upload	4 years ago
morizin	c2d5087eae	Add files via upload	4 years ago
Ross Wightman	9a1bd358c7	Merge pull request #571 from normster/augmix-fix ... Enable uniform augmentation magnitude sampling and set AugMix default	4 years ago
Norman Mu	79640fcc1f	Enable uniform augmentation magnitude sampling and set AugMix default	4 years ago
Ross Wightman	c1cf9712fc	Add updated EfficientNet-V2S weights, 83.8 @ 384x384 test. Add PyTorch trained EfficientNet-B4 weights, 83.4 @ 384x384 test. Tweak non TF EfficientNet B1-B4 train/test res scaling.	4 years ago
Ross Wightman	e8a64fb881	Test input size for efficientnet_v2s was wrong in last results run	4 years ago