19 Commits (a16ea1e3558890ba2ac9b11a85c56a4e36c7afb9)

Author	SHA1	Message	Date
Ross Wightman	ff0f709c20	Testing TFDS shuffle across epochs	3 years ago
Ross Wightman	cad170e494	Merge remote-tracking branch 'origin/norm_norm_norm' into bits_and_tpu	3 years ago
Ross Wightman	55adfbeb8d	Add commented code to increase open file limit via Python (for TFDS dataset building)	3 years ago
Ross Wightman	809c7bb1ec	Merge remote-tracking branch 'origin/master' into bits_and_tpu	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
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	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	25d52ea71d	Merge remote-tracking branch 'origin/fixes_bce_regnet' into bits_and_tpu	3 years ago
Ross Wightman	b57a03bd0d	Merge branch 'master' into bits_and_tpu	4 years ago
Ross Wightman	91ab0b6ce5	Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go...	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	d53e91218e	Fix tf.data options setting for newer TF versions	4 years ago
Ross Wightman	938716c753	Fix import issue, use devenv for dist info in parser_tfds	4 years ago
Ross Wightman	f42f1df26c	Improve evenness of per-worker split for validation set with TFDS	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	5d4c3d0af3	Add enhanced ParserImageInTar that can read images from tars within tars, folders with multiple tars, etc. Additional comment cleanup.	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