22 Commits (229ac6b8d88e3523db741ba74b5ac9da5b30720c)

Author	SHA1	Message	Date
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	59a3409182	Update README.md	3 years ago
Ross Wightman	3581affb77	Update train.py with some flags related to scheduler tweaks, fix best checkpoint bug.	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	b76b48e8e9	Update optimizer creation for master optimizer changes	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	847b4af144	Update README.md	4 years ago
Ross Wightman	5c5cadfe4c	Update README.md	4 years ago
Ross Wightman	ee2b8f49ee	Update README.md	4 years ago
Ross Wightman	cc870df7b8	Update README.md	4 years ago
Ross Wightman	6b2d9c2660	Another bits/README.md update	4 years ago
Ross Wightman	c3db5f5801	Worker hack for TFDS eval, add TPU env var setting.	4 years ago
Ross Wightman	f411724de4	Fix checkpoint delete issue. Add README about bits and initial Pytorch XLA usage on TPU-VM. Add some FIXMEs and fold train_cfg into train_state by default.	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	5b9c69e80a	Add basic training resume based on legacy code	4 years ago
Ross Wightman	72ca831dd4	Back to using strings for the enum translation, forgot about import dep	4 years ago
Ross Wightman	cbd4ee737f	Fix model init for XLA, remove some prints.	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	938716c753	Fix import issue, use devenv for dist info in parser_tfds	4 years ago
Ross Wightman	76de984a5f	Fix some bugs with XLA support, logger, add hacky xla dist launch script since torch.dist.launch doesn't work	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