728 Commits (d9b0b3d60f5a49403799886583b58ff2bd9d8b43)

Author	SHA1	Message	Date
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