6 Commits (4210d922d2b478263a766c9b89f9898109a583d9)

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