You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ross Wightman 56ed0a0b63 Merge branch 'vit_and_bit_test_fixes' into bits_and_tpu		3 years ago
..
bits	Update README.md	3 years ago
data	Merge branch 'master' into bits_and_tpu	3 years ago
loss	asymmetric_loss	4 years ago
models	Add gMLP-S weights, 79.6 top-1	3 years ago
optim	Fix adabelief imports, remove prints, preserve memory format is the default arg for zeros_like	3 years ago
scheduler	Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes.	3 years ago
utils	Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go...	3 years ago
__init__.py	Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models	3 years ago
version.py	Bump version 0.4.12	3 years ago