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
cbd4ee737f
Fix model init for XLA, remove some prints.
|
4 years ago |
| .. |
|
bits
|
Fix model init for XLA, remove some prints.
|
4 years ago |
|
data
|
Merge branch 'master' into bits_and_tpu
|
4 years ago |
|
loss
|
asymmetric_loss
|
4 years ago |
|
models
|
Fix drop/drop_path arg on MLP-Mixer model. Fix #641
|
4 years ago |
|
optim
|
Fix adabelief imports, remove prints, preserve memory format is the default arg for zeros_like
|
4 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.
|
4 years ago |
|
utils
|
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 |
|
__init__.py
|
Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models
|
4 years ago |
|
version.py
|
Update README, tweak fine-tune effv2 model names.
|
4 years ago |
