71 Commits (bits_and_tpu)

Author	SHA1	Message	Date
Ross Wightman	ab16a358bb	Add log and continue handler for WDS errors, fix args.num_gpu for validation script fallback	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
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	07693f81b0	Validation fix since we don't have multi-GPU DataParallel support yet	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	e0b3a3fab3	Make test-pooling flag for validate.py opt in	3 years ago
Ross Wightman	25d52ea71d	Merge remote-tracking branch 'origin/fixes_bce_regnet' into bits_and_tpu	3 years ago
Ross Wightman	5d6983c462	Batch validate a list of files if model is a text file with model per line	3 years ago
Ross Wightman	f2e14685a8	Add force-cpu flag for train/validate, fix CPU fallback for device init, remove old force cpu flag for EMA model weights	3 years ago
Ross Wightman	40457e5691	Transforms, augmentation work for bits, add RandomErasing support for XLA (pushing into transforms), revamp of transform/preproc config, etc ongoing...	3 years ago
Ross Wightman	91ab0b6ce5	Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go...	3 years ago
Ross Wightman	74d2829341	Merge branch 'master' into bits_and_tpu	4 years ago
Ross Wightman	aa92d7b1c5	Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes.	4 years ago
Ross Wightman	12d9a6d4d2	First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working	4 years ago
Ross Wightman	a04427d8ce	Add _in22k to bulk validate filter	4 years ago
Ross Wightman	a5310a3451	Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update	4 years ago
Ross Wightman	d584e7f617	Support for huggingface hub via create_model and default_cfgs. ... * improve consistency of model creation helper fns * add comments to some of the model helpers * support passing external default_cfgs so they can be sourced from hub	4 years ago
Ross Wightman	2db2d87ff7	Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.	4 years ago
Ross Wightman	d8e69206be	Merge pull request #419 from rwightman/byob_vgg_models ... More models, GPU-Efficient Nets, RepVGG, classic VGG, and flexible Byob backbone.	4 years ago
Ross Wightman	0356e773f5	Default to native PyTorch AMP instead of APEX amp. Too many APEX issues cropping up lately.	4 years ago
Ross Wightman	b4e216e377	Fix a few small things.	4 years ago
Csaba Kertesz	5114c214fc	Change the Python interpreter to Python 3.x in the scripts	4 years ago
Ross Wightman	2a8c4dc63b	Add validation script update for using test_input_size in model default_cfgs	4 years ago
Ross Wightman	38d8f67570	Fix potential issue with change to num_classes arg in train/validate.py defaulting to None (rely on model def / default_cfg)	4 years ago
Ross Wightman	855d6cc217	More dataset work including factories and a tensorflow datasets (TFDS) wrapper ... * Add parser/dataset factory methods for more flexible dataset & parser creation * Add dataset parser that wraps TFDS image classification datasets * Tweak num_classes handling bug for 21k models * Add initial deit models so they can be benchmarked in next csv results runs	4 years ago
Ross Wightman	59ec7e6a53	Merge branch 'master' into imagenet21k_datasets_more	4 years ago
Csaba Kertesz	e42b140ade	Add --input-size option to scripts to specify full input dimensions from command-line	4 years ago
Ross Wightman	ce69de70d3	Add 21k weight urls to vision_transformer. Cleanup feature_info for preact ResNetV2 (BiT) models	4 years ago
Ross Wightman	de6046e213	Initial commit for dataset / parser reorg to support additional datasets / types	4 years ago
Matthijs Hollemans	8ffdc5910a	test_time_pool would be set to a non-False value even if test-time pooling is not available	4 years ago
Ross Wightman	e8ca45854c	More models in sotabench, more control over sotabench run, dataset filename extraction consistency	4 years ago
Ross Wightman	751b0bba98	Add global_pool (--gp) arg changes to allow passing 'fast' easily for train/validate to avoid channels_last issue with AdaptiveAvgPool	4 years ago
Ross Wightman	9c297ec67d	Cleanup Apex vs native AMP scaler state save/load. Cleanup CheckpointSaver a bit.	4 years ago
Ross Wightman	c2cd1a332e	Improve torch amp support and add channels_last support for train/validate scripts	4 years ago
Ross Wightman	ad150e7018	Update results csv file rank/diff script and small validate script tweak for batch validation	4 years ago
Ross Wightman	ecdeb470f2	Cleanup real labels in validate.py, no need to track original, will do that via exteran script	4 years ago
Ross Wightman	c53ec33ae0	Add synset/label indices for results generation. Add 'valid labels' to validation script to support imagenet-a/r label subsets properly.	4 years ago
Ross Wightman	7995295968	Merge branch 'logger' into features. Change 'logger' to '_logger'.	4 years ago
Ross Wightman	1998bd3180	Merge branch 'feature/AB/logger' of https://github.com/antoinebrl/pytorch-image-models into logger	4 years ago
Ross Wightman	6c17d57a2c	Fix some attributions, add copyrights to some file docstrings	4 years ago
Ross Wightman	4e61c6a12d	Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript	4 years ago
Ross Wightman	d23a2697d0	Working on feature extraction, interfaces refined, a number of models working, some in progress.	4 years ago
Ross Wightman	d3ee3de96a	Update validation script first batch prime and clear cuda cache between multi-model runs	4 years ago
Antoine Broyelle	78fa0772cc	Leverage python hierachical logger ... with this update one can tune the kind of logs generated by timm but training and inference traces are unchanged	4 years ago
Ross Wightman	88129b2569	Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.	5 years ago
Ross Wightman	eb7653614f	Monster commit, activation refactor, VoVNet, norm_act improvements, more ... * refactor activations into basic PyTorch, jit scripted, and memory efficient custom auto * implement hard-mish, better grad for hard-swish * add initial VovNet V1/V2 impl, fix #151 * VovNet and DenseNet first models to use NormAct layers (support BatchNormAct2d, EvoNorm, InplaceIABN) * Wrap IABN for any models that use it * make more models torchscript compatible (DPN, PNasNet, Res2Net, SelecSLS) and add tests	5 years ago
Ross Wightman	02a30411ad	Replace fp16 with amp support for validate.py script	5 years ago
Ross Wightman	13cf68850b	Remove poorly named metrics from torch imagenet example origins. Use top1/top5 in csv output for consistency with existing validation results files, acc elsewhere. Fixes #111	5 years ago
Ross Wightman	53c47479c4	Batch validation batch size adjustment, tweak L2 crop pct	5 years ago