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
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 |
.. |
__init__.py
|
Remove commented code, add more consistent seed fn
|
4 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
|
Reorg of utils into separate modules
|
4 years ago |
log.py
|
Reorg of utils into separate modules
|
4 years ago |
metrics.py
|
Fix topn metric view regression on PyTorch 1.7
|
4 years ago |
misc.py
|
Reorg of utils into separate modules
|
4 years ago |
model.py
|
Add GIST to docstring for quick access
|
4 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 |