1015 Commits (bits_and_tpu)

Author	SHA1	Message	Date
Ross Wightman	ef57561d51	Fix some TPU (XLA) issues with swin transformer v2	3 years ago
Ross Wightman	ab16a358bb	Add log and continue handler for WDS errors, fix args.num_gpu for validation script fallback	3 years ago
Ross Wightman	7eeaf521a0	use gopen in wds to open info file in case it's at a url/gs location	3 years ago
Ross Wightman	229ac6b8d8	Fix alternate label handling in WDS parser to skip invalid alt labels	3 years ago
Ross Wightman	a444d4b891	Add alternative label support to WDS for imagenet22k/12k split, add 21k/22k/12k indices filters to results/	3 years ago
Ross Wightman	da2796ae82	Add webdataset (WDS) support, update TFDS to make some naming in parsers more similar. Fix workers=0 compatibility. Add ImageNet22k/12k synset defs.	3 years ago
Ross Wightman	3fce010ca8	Merge remote-tracking branch 'origin/norm_norm_norm' into bits_and_tpu	3 years ago
Ross Wightman	0557c8257d	Fix bug introduced in non layer_decay weight_decay application. Remove debug print, fix arg desc.	3 years ago
Ross Wightman	15cc9eae3e	Fix Swin v2 tuple type hint	3 years ago
Ross Wightman	bb85b09d2a	swin v2 fixup for latest changes on norm_norm_norm / bits_and_tpu branch	3 years ago
Ross Wightman	10fa42b143	Merge branch 'ChristophReich1996-master' into bits_and_tpu	3 years ago
Ross Wightman	c639a86c67	Change TFDS default to full re-shuffle (init) each epoch (for now)	3 years ago
Ross Wightman	a16ea1e355	Merge remote-tracking branch 'origin/norm_norm_norm' into bits_and_tpu	3 years ago
Ross Wightman	372ad5fa0d	Significant model refactor and additions: ... * All models updated with revised foward_features / forward_head interface * Vision transformer and MLP based models consistently output sequence from forward_features (pooling or token selection considered part of 'head') * WIP param grouping interface to allow consistent grouping of parameters for layer-wise decay across all model types * Add gradient checkpointing support to a significant % of models, especially popular architectures * Formatting and interface consistency improvements across models * layer-wise LR decay impl part of optimizer factory w/ scale support in scheduler * Poolformer and Volo architectures added	3 years ago
Ross Wightman	1420c118df	Missed comitting outstanding changes to default_cfg keys and test exclusions for swin v2	3 years ago
Ross Wightman	c6e4b7895a	Swin V2 CR impl refactor. ... * reformat and change some naming so closer to existing timm vision transformers * remove typing that wasn't adding clarity (or causing torchscript issues) * support non-square windows * auto window size adjust from image size * post-norm + main-branch no	3 years ago
Christoph Reich	67d140446b	Fix bug in classification head	3 years ago
Christoph Reich	29add820ac	Refactor (back to relative imports)	3 years ago
Christoph Reich	74a04e0016	Add parameter to change normalization type	3 years ago
Christoph Reich	2a4f6c13dd	Create model functions	3 years ago
Christoph Reich	87b4d7a29a	Add get and reset classifier method	3 years ago
Christoph Reich	ff5f6bcd6c	Check input resolution	3 years ago
Christoph Reich	81bf0b4033	Change parameter names to match Swin V1	3 years ago
Christoph Reich	f227b88831	Add initials (CR) to model and file	3 years ago
Christoph Reich	90dc74c450	Add code from https://github.com/ChristophReich1996/Swin-Transformer-V2 and change docstring style to match timm	3 years ago
Ross Wightman	2c3870e107	semobilevit_s for good measure	3 years ago
Ross Wightman	bcaeb91b03	Version to 0.6.0, possible interface incompatibilities vs 0.5.x	3 years ago
Ross Wightman	58ba49c8ef	Add MobileViT models (w/ ByobNet base). Close #1038.	3 years ago
Ross Wightman	066e490605	Merge branch 'norm_norm_norm' into bits_and_tpu	3 years ago
Ross Wightman	5f81d4de23	Move DeiT to own file, vit getting crowded. Working towards fixing #1029, make pooling interface for transformers and mlp closer to convnets. Still working through some details...	3 years ago
ayasyrev	cf57695938	sched noise dup code remove	3 years ago
Ross Wightman	95cfc9b3e8	Merge remote-tracking branch 'origin/master' into norm_norm_norm	3 years ago
Ross Wightman	abc9ba2544	Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.	3 years ago
Ross Wightman	07379c6d5d	Add vit_base2_patch32_256 for a model between base_patch16 and patch32 with a slightly larger img size and width	3 years ago
Ross Wightman	447677616f	version 0.5.5	3 years ago
Ross Wightman	83b40c5a58	Last batch of small model weights (for now). mobilenetv3_small 050/075/100 and updated mnasnet_small with lambc/lamb optimizer.	3 years ago
Mi-Peng	cdcd0a92ca	fix lars	3 years ago
Ross Wightman	1aa617cb3b	Add AvgPool2d anti-aliasing support to ResNet arch (as per OpenAI CLIP models), add a few blur aa models as well	3 years ago
Ross Wightman	f0f9eccda8	Add --fuser arg to train/validate/benchmark scripts to select jit fuser type	3 years ago
Ross Wightman	010b486590	Add Dino pretrained weights (no head) for vit models. Add support to tests and helpers for models w/ no classifier (num_classes=0 in pretrained cfg)	3 years ago
Ross Wightman	738a9cd635	unbiased=False for torch.var_mean path of ConvNeXt LN. Fix #1090	3 years ago
Ross Wightman	e0c4eec4b6	Default conv_mlp to False across the board for ConvNeXt, causing issues on more setups than it's improving right now...	3 years ago
Ross Wightman	b669f4a588	Add ConvNeXt 22k->1k fine-tuned and 384 22k-1k fine-tuned weights after testing	3 years ago
Ross Wightman	e967c72875	Update REAMDE.md. Sneak in g/G (giant / gigantic?) ViT defs from scaling paper	3 years ago
Ross Wightman	9ca3437178	Add some more small model weights lcnet, mnas, mnv2	3 years ago
Ross Wightman	fa6463c936	Version 0.5.4	3 years ago
Ross Wightman	fa81164378	Fix stem width for really small mobilenetv3 arch defs	3 years ago
Ross Wightman	edd3d73695	Add missing dropout for head reset in ConvNeXt default head	3 years ago
Ross Wightman	b093dcb46d	Some convnext cleanup, remove in place mul_ for gamma, breaking symbolic trace, cleanup head a bit...	3 years ago
Ross Wightman	18934debc5	Add initial ConvNeXt impl (mods of official code)	3 years ago
Ross Wightman	656757d26b	Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.	3 years ago
Ross Wightman	ccfeb06936	Fix out_indices handling breakage, should have left as per vgg approach.	3 years ago
Ross Wightman	a9f91483a6	Fix #1078, DarkNet has 6 feature maps. Make vgg and darknet out_indices handling/comments equivalent	3 years ago
Ross Wightman	c21b21660d	visformer supports spatial feat map, update pool_size in pretrained cfg to match	3 years ago
Ross Wightman	9c11dfd9cb	Fix fbnetv3 pretrained cfg changes	3 years ago
Ross Wightman	1406cddc2e	FBNetV3 timm trained weights added for b/d/g variants. Update version to 0.5.2 for pypi release.	3 years ago
Ross Wightman	02ae11e526	Leaving repeat aug sampler indices as tensor thrashes worker shared process memory	3 years ago
Ross Wightman	4df51f3932	Add lcnet_100 and mnasnet_small weights	3 years ago
Ross Wightman	5ccf682a8f	Remove deprecated bn-tf train arg and create_model handler. Add evos/evob models back into fx test filter until norm_norm_norm branch merged.	3 years ago
Ross Wightman	b9a715c86a	Add more small model defs for MobileNetV3/V2/LCNet	3 years ago
Ross Wightman	b27c21b09a	Update drop_path and drop_block (fast impl) to be symbolically traceable, slightly faster	3 years ago
Ross Wightman	214c84a235	Disable use of timm nn.Linear wrapper since AMP autocast + torchscript use appears fixed	3 years ago
Ross Wightman	72b57163d1	Merge branch 'master' of https://github.com/mrT23/pytorch-image-models into mrT23-master	3 years ago
Ross Wightman	de5fa791c6	Merge branch 'master' into norm_norm_norm	3 years ago
Ross Wightman	26ff57f953	Add more small model defs for MobileNetV3/V2/LCNet	3 years ago
Hyeongchan Kim	a0b2657497	Use `torch.repeat_interleave()` to generate repeated indices faster (#1058) ... * update: use numpy to generate repeated indices faster * update: use torch.repeat_interleave() instead of np.repeat() * refactor: remove unused import, numpy * refactor: torch.range to torch.arange * update: tensor to list before appending the extra samples * update: concatenate the paddings with torch.cat	3 years ago
Ross Wightman	450ac6a0f5	Post merge tinynet fixes for pool_size, feature extraction	3 years ago
Ross Wightman	a04164cd75	Merge branch 'tinynet' of https://github.com/rsomani95/pytorch-image-models into rsomani95-tinynet	3 years ago
Ross Wightman	7eb7e73216	File will not stay deleted	3 years ago
Ross Wightman	0012bf7fb5	Merge branch 'norm_norm_norm' into bits_and_tpu	3 years ago
Ross Wightman	8a93ce6ee3	Fix regnetv/w tests, refactor regnet generator code a bit	3 years ago
Ross Wightman	cbc4f33220	Merge branch 'norm_norm_norm' into bits_and_tpu	3 years ago
Ross Wightman	4dec8c8087	Fix skip path regression for updated EfficientNet and RegNet def. Add Pre-Act RegNet support (experimental). Remove BN-TF flag. Add efficientnet_b0_g8_gn model.	3 years ago
Ross Wightman	a52a614475	Remove layer experiment which should not have been added	3 years ago
Ross Wightman	d829858550	Significant norm update ... * ConvBnAct layer renamed -> ConvNormAct and ConvNormActAa for anti-aliased * Significant update to EfficientNet and MobileNetV3 arch to support NormAct layers and grouped conv (as alternative to depthwise) * Update RegNet to add Z variant * Add Pre variant of XceptionAligned that works with NormAct layers * EvoNorm matches bits_and_tpu branch for merge	3 years ago
Ross Wightman	ab49d275de	Significant norm update ... * ConvBnAct layer renamed -> ConvNormAct and ConvNormActAa for anti-aliased * Significant update to EfficientNet and MobileNetV3 arch to support NormAct layers and grouped conv (as alternative to depthwise) * Update RegNet to add Z variant * Add Pre variant of XceptionAligned that works with NormAct layers * EvoNorm matches bits_and_tpu branch for merge	3 years ago
Rahul Somani	31bcd36e46	add tinynet models	3 years ago
KAI ZHAO	b4b8d1ec18	fix hard-coded strides	3 years ago
Ross Wightman	57fca2b5b2	Fix c16_evos stem / first conv setup	3 years ago
Ross Wightman	1f54a1fff7	Add C16 and E8 EvoNormS0 configs for RegNetZ BYOB nets	3 years ago
Ross Wightman	4d7a5544f7	Remove inplace sigmoid for consistency with other impl	3 years ago
Ross Wightman	88a5b54802	A few small evonorm tweaks for convergence comparisons	3 years ago
Ross Wightman	66daee4f31	Last change wasn't complete, missed adding full evo_norm changeset	3 years ago
Ross Wightman	7bbbd5ef1b	EvoNorm and GroupNormAct options for debugging TPU / XLA concerns	3 years ago
Ross Wightman	ff0f709c20	Testing TFDS shuffle across epochs	3 years ago
Ross Wightman	69e90dcd8c	Merge branch 'norm_norm_norm' into bits_and_tpu	3 years ago
Ross Wightman	d04f2f1377	Update drop_path and drop_block (fast impl) to be symbolically traceable, slightly faster	3 years ago
Ross Wightman	820ae9925e	Fix load_state_dict to handle None ema entries	3 years ago
Ross Wightman	0e212e8fe5	Merge remote-tracking branch 'origin/norm_norm_norm' into bits_and_tpu	3 years ago
Ross Wightman	834a9ec721	Disable use of timm nn.Linear wrapper since AMP autocast + torchscript use appears fixed	3 years ago
Ross Wightman	cad170e494	Merge remote-tracking branch 'origin/norm_norm_norm' into bits_and_tpu	3 years ago
Ross Wightman	78912b6375	Updated EvoNorm implementations with some experimentation. Add FilterResponseNorm. Updated RegnetZ and ResNetV2 model defs for trials.	3 years ago
Ross Wightman	55adfbeb8d	Add commented code to increase open file limit via Python (for TFDS dataset building)	3 years ago
talrid	c11f4c3218	support CNNs	3 years ago
mrT23	d6701d8a81	Merge branch 'rwightman:master' into master	3 years ago
qwertyforce	ccb3815360	update arxiv link	3 years ago
Ross Wightman	3dc71695bf	Merge pull request #989 from martinsbruveris/feat/resmlp-dino ... Added DINO pretrained ResMLP models.	3 years ago
Ross Wightman	480c676ffa	Fix FX breaking assert in evonorm	3 years ago
Martins Bruveris	85c5ff26d7	Added DINO pretrained ResMLP models.	3 years ago
Ross Wightman	d633a014e6	Post merge cleanup. Fix potential security issue passing kwargs directly through to serialized web data.	3 years ago
Nathan Raw	b18c9e323b	Update helpers.py	3 years ago
Nathan Raw	308d0b9554	Merge branch 'master' into hf-save-and-push	3 years ago
Ross Wightman	f0507f6da6	Fix k_decay default arg != 1.0 in poly scheduler	3 years ago
talrid	41559247e9	use_ml_decoder_head	3 years ago
Ross Wightman	1f53db2ece	Updated lamhalobotnet weights, 81.5 top-1	3 years ago
Ross Wightman	15ef108eb4	Add better halo2botnet50ts weights, 82 top-1 @ 256	3 years ago
Ross Wightman	809c7bb1ec	Merge remote-tracking branch 'origin/master' into bits_and_tpu	3 years ago
Ross Wightman	734b2244fe	Add RegNetZ-D8 (83.5 @ 256, 84 @ 320) and RegNetZ-E8 (84.5 @ 256, 85 @ 320) weights. Update names of existing RegZ models to include group size.	3 years ago
Ross Wightman	93cc08fdc5	Make evonorm variables 1d to match other PyTorch norm layers, will break weight compat for any existing use (likely minimal, easy to fix).	3 years ago
Ross Wightman	af607b75cc	Prep a set of ResNetV2 models with GroupNorm, EvoNormB0, EvoNormS0 for BN free model experiments on TPU and IPU	3 years ago
Ross Wightman	c976a410d9	Add ResNet-50 w/ GN (resnet50_gn) and SEBotNet-33-TS (sebotnet33ts_256) model defs and weights. Update halonet50ts weights w/ slightly better variant in1k val, more robust to test sets.	3 years ago
Ross Wightman	f2006b2437	Cleanup qkv_bias cat in beit model so it can be traced	3 years ago
Ross Wightman	1076a65df1	Minor post FX merge cleanup	3 years ago
Ross Wightman	32c9937dec	Merge branch 'fx-feature-extract-new' of https://github.com/alexander-soare/pytorch-image-models into alexander-soare-fx-feature-extract-new	3 years ago
Ross Wightman	78b36bf46c	Places365 doesn't exist in some still used torchvision version	3 years ago
Alexander Soare	65d827c7a6	rename notrace registration and standardize trace_utils imports	3 years ago
Ross Wightman	9b2daf2a35	Add ResNeXt-50 weights 81.1 top-1 @ 224, 82 @ 288 with A1 'high aug' recipe	3 years ago
Ross Wightman	9b5d6dc7e2	Merge branch 'add-vit-b8' of https://github.com/martinsbruveris/pytorch-image-models into martinsbruveris-add-vit-b8	3 years ago
Ross Wightman	cfa414cad2	Matching two bits_and_tpu changes for TFDs wrapper ... * change 'samples' -> 'examples' for tfds wrapper to match tfds naming * add class_to_idx for image classification datasets in tfds wrapper	3 years ago
Martins Bruveris	5220711d87	Added B/8 models to ViT.	3 years ago
Alexander Soare	0262a0e8e1	fx ready for review	3 years ago
Ross Wightman	871cef4198	version 0.5.1	3 years ago
Ross Wightman	4f338556d8	Fixes and improvements for metrics, tfds parser, loader / transform handling ... * add back ability to create transform with loader * change 'samples' -> 'examples' for tfds wrapper to match tfds naming * add support for specifying feature names for input and target in tfds wrapper * add class_to_idx for image classification datasets in tfds wrapper * add accumulate_type to avg meters and metrics to allow float32 or float64 accumulation control with lower prec data * minor cleanup, log output rate prev and avg	3 years ago
Alexander Soare	d2994016e9	Add try/except guards	3 years ago
Alexander Soare	b25ff96768	wip - pre-rebase	3 years ago
Alexander Soare	e051dce354	Make all models FX traceable	3 years ago
Alexander Soare	cf4561ca72	Add FX based FeatureGraphNet capability	3 years ago
Alexander Soare	0149ec30d7	wip - attempting to rebase	3 years ago
Alexander Soare	02c3a75a45	wip - make it possible to use fx graph in train and eval mode	3 years ago
Alexander Soare	bc3d4eb403	wip -rebase	3 years ago
Alexander Soare	ab3ac3f25b	Add FX based FeatureGraphNet capability	3 years ago
Ross Wightman	d9b0b3d60f	device arg wasn't removed from PrefetcherCuda instantiation of RE	3 years ago
Ross Wightman	80ca078aed	Fix a few bugs and formatting/naming issues ... * Pass optimizer resume flag through to checkpoint / updater restore. Related to #961 but not clear how relates to crash. * Rename monitor step args, cleanup handling of step_end_idx vs num_steps for consistent log output in either case * Resume from proper epoch (ie next epoch relative to checkpoint)	3 years ago
Ross Wightman	406c486ba2	Merge remote-tracking branch 'origin/more_datasets' into bits_and_tpu	3 years ago
Ross Wightman	9ec3210c2d	More TFDS parser cleanup, support improved TFDS even_split impl (on tfds-nightly only currently).	3 years ago
Ross Wightman	ba65dfe2c6	Dataset work ... * support some torchvision datasets * improvements to TFDS wrapper for subsplit handling (fix #942), shuffle seed * add class-map support to train (fix #957)	3 years ago
Ross Wightman	59a3409182	Update README.md	3 years ago
Ross Wightman	ddc29da974	Add ResNet101 and ResNet152 weights from higher aug RSB recipes. 81.93 and 82.82 top-1 at 224x224.	3 years ago
Ross Wightman	b328e56f49	Update eca_halonext26ts weights to a better set	3 years ago
Ross Wightman	2ddef942b9	Better fix for #954 that doesn't break torchscript, pull torch._assert into timm namespace when it exists	3 years ago
Ross Wightman	4f0f9cb348	Fix #954 by bringing traceable _assert into timm to allow compat w/ PyTorch < 1.8	3 years ago
Ross Wightman	a41de1f666	Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix #738.	3 years ago
Ross Wightman	ed41d32637	Add repr to auto_augment and random_erasing impl	3 years ago
Ross Wightman	ae72d009fa	Add weights for lambda_resnet50ts, halo2botnet50ts, lamhalobotnet50ts, updated halonet50ts	3 years ago
Ross Wightman	a45186a6e8	Merge remote-tracking branch 'origin/master' into bits_and_tpu	3 years ago
Ross Wightman	b745d30a3e	Fix formatting of last commit	3 years ago
Ross Wightman	3478f1d7f1	Traceability fix for vit models for some experiments	3 years ago
Ross Wightman	f658a72e72	Cleanup re-use of Dropout modules in Mlp modules after some twitter feedback :p	3 years ago
Thomas Viehmann	f805ba86d9	use .unbind instead of explicitly listing the indices	3 years ago
Ross Wightman	57992509f9	Fix some formatting in utils/model.py	3 years ago
Ross Wightman	0fe4fd3f1f	add d8 and e8 regnetz models with group size 8	3 years ago
Ross Wightman	690f31d02d	Post merge cleanup, restore previous unwrap fn	3 years ago
Ross Wightman	3b6ba76126	Merge remote-tracking branch 'origin/master' into bits_and_tpu	3 years ago
Ross Wightman	25e7c8c5e5	Update broken resnetv2_50 weight url, add resnetv1_101 a1h recipe weights for 224x224 train	3 years ago
Ross Wightman	b6caa356d2	Fixed eca_botnext26ts_256 weights added, 79.27	3 years ago
Ross Wightman	c02334d9fa	Add weights for regnetz_d and haloregnetz_c, update regnetz_c weights. Add commented PyTorch XLA code for halo attention	3 years ago
Ross Wightman	02daf2ab94	Add option to include relative pos embedding in the attention scaling as per references. See discussion #912	3 years ago
masafumi	047a5ec05f	Fix bugs that Mixup does not work device=cpu	3 years ago
Ross Wightman	cd34913278	Remove some outdated comments, botnet networks working great now.	3 years ago
Ross Wightman	6ed4cdccca	Update lambda_resnet26t weights with better set	3 years ago
ICLR Author	44d6d51668	Add ConvMixer	3 years ago
Ross Wightman	a85df34993	Update lambda_resnet26rpt weights to 78.9, add better halonet26t weights at 79.1 with tweak to attention dim	3 years ago
Ross Wightman	b544ad4d3f	regnetz model default cfg tweaks	3 years ago
Ross Wightman	e5da481073	Small post-merge tweak for freeze/unfreeze, add to __init__ for utils	3 years ago
Ross Wightman	5ca72dcc75	Merge branch 'freeze-functionality' of https://github.com/alexander-soare/pytorch-image-models into alexander-soare-freeze-functionality	3 years ago
Ross Wightman	e2b8d44ff0	Halo, bottleneck attn, lambda layer additions and cleanup along w/ experimental model defs ... * align interfaces of halo, bottleneck attn and lambda layer * add qk_ratio to all of above, control q/k dim relative to output dim * add experimental haloregnetz, and trionet (lambda + halo + bottle) models	3 years ago
Alexander Soare	431e60c83f	Add acknowledgements for freeze_batch_norm inspiration	3 years ago
Ross Wightman	fbf59c04ee	Change crop ratio on correct resnet50 variant.	3 years ago
Ross Wightman	ae1ff5792f	Clean a1/a2/3 rsb _0 checkpoints properly, fix v2 loading.	3 years ago
Ross Wightman	93901e992f	Version bump to 0.5.0 for pending release post RSB and ATTN updates	3 years ago
Ross Wightman	da0d39bedd	Update default crop_pct for byoanet	3 years ago
Ross Wightman	cc9bedf373	Add initial ResNet Strikes Back weights for ResNet50 and ResNetV2-50 models	3 years ago
Ross Wightman	64495505b7	Add updated lambda resnet26 and botnet26 checkpoints with fixes applied	3 years ago
Ross Wightman	b2094f4ee8	support bits checkpoints in avg/load	3 years ago
Ross Wightman	007bc39323	Some halo and bottleneck attn code cleanup, add halonet50ts weights, use optimal crop ratios	3 years ago
Alexander Soare	65c3d78b96	Freeze unfreeze functionality finalized. Tests added	3 years ago
Alexander Soare	0cb8ea432c	wip	3 years ago
Ross Wightman	1fdc7af8fd	Merge remote-tracking branch 'origin/fixes_bce_regnet' into bits_and_tpu	3 years ago
Ross Wightman	b1c2e3eb92	Match rel_pos_indices attr rename in conv branch	3 years ago
Ross Wightman	b49630a138	Add relative pos embed option to LambdaLayer, fix last transpose/reshape.	3 years ago
Ross Wightman	d657e2cc0b	Remove dead code line from efficientnet	3 years ago
Ross Wightman	0ca687f224	Make 'regnetz' model experiments closer to actual RegNetZ, bottleneck expansion, expand from in_chs, no shortcut on stride 2, tweak model sizes	3 years ago
leondgarse	51eaf9360d	Remove a duplicate layer creation in byobnet.py ... `self.conv2_kxk` is repeated in `byobnet.py`. Remove the duplicate code.	3 years ago
Ross Wightman	b81e79aae9	Fix bottleneck attn transpose typo, hopefully these train better now..	3 years ago
Ross Wightman	80075b0b8a	Add worker_seeding arg to allow selecting old vs updated data loader worker seed for (old) experiment repeatability	3 years ago
Ross Wightman	52c481ea8e	Merge remote-tracking branch 'origin/fixes_bce_regnet' into bits_and_tpu	3 years ago
Ross Wightman	6478bcd02c	Fix regnetz_d conv layer name, use inception mean/std	3 years ago
Ross Wightman	25d52ea71d	Merge remote-tracking branch 'origin/fixes_bce_regnet' into bits_and_tpu	3 years ago
Ross Wightman	0387e6057e	Update binary cross ent impl to use thresholding as an option (convert soft targets from mixup/cutmix to 0, 1)	3 years ago
Ross Wightman	f8a63a3b71	Add worker_init_fn to loader for numpy seed per worker	3 years ago
Ross Wightman	515121cca1	Use reshape instead of view in std_conv, causing issues in recent PyTorch in channels_last	3 years ago
Ross Wightman	da06cc61d4	ResNetV2 seems to work best without zero_init residual	3 years ago
Ross Wightman	8e11da0ce3	Add experimental RegNetZ(ish) models for training / perf trials.	3 years ago
Alexander Soare	6bbc50beb4	make it possible to provide norm_layer via create_model	3 years ago
nateraw	adcb74f87f	🎨 Import load_state_dict_from_url directly	3 years ago
nateraw	e65a2cba3d	🎨 cleanup and add a couple comments	3 years ago
nateraw	2b6ade24b3	🎨 write model card to enable inference	3 years ago
Ross Wightman	cf5ac2800c	BotNet models were still off, remove weights for bad configs. Add good SE-HaloNet33-TS weights.	3 years ago
Ross Wightman	24720abe3b	Merge branch 'master' into attn_update	3 years ago
Ross Wightman	1c9284c640	Add BeiT 'finetuned' 1k weights and pretrained 22k weights, pretraining specific (masked) model excluded for now	3 years ago