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
a45186a6e8
Merge remote-tracking branch 'origin/master' into bits_and_tpu
|
3 years ago |
.. |
__init__.py
|
Small post-merge tweak for freeze/unfreeze, add to __init__ for utils
|
3 years ago |
agc.py
|
Initial AGC impl. Still testing.
|
4 years ago |
checkpoint_saver.py
|
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 |
clip_grad.py
|
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 |
cuda.py
|
Initial AGC impl. Still testing.
|
4 years ago |
distributed.py
|
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 |
jit.py
|
…
|
|
log.py
|
…
|
|
metrics.py
|
Tweak accuracy topk safety. Fix #807
|
3 years ago |
misc.py
|
…
|
|
model.py
|
Merge remote-tracking branch 'origin/master' into bits_and_tpu
|
3 years ago |
model_ema.py
|
Add separate set and update method to ModelEmaV2
|
4 years ago |
random.py
|
Remove commented code, add more consistent seed fn
|
4 years ago |
summary.py
|
Make wandb optional
|
4 years ago |