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
bd6d377c74
Merge remote-tracking branch 'origin/master' into bits_and_tpu
|
2 years ago |
| .. |
|
__init__.py
|
Add bulk_runner script and updates to benchmark.py and validate.py for better error handling in bulk runs (used for benchmark and validation result runs). Improved batch size decay stepping on retry...
|
2 years ago |
|
agc.py
|
Initial AGC impl. Still testing.
|
3 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.
|
3 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.
|
3 years ago |
|
cuda.py
|
Initial AGC impl. Still testing.
|
3 years ago |
|
decay_batch.py
|
Add bulk_runner script and updates to benchmark.py and validate.py for better error handling in bulk runs (used for benchmark and validation result runs). Improved batch size decay stepping on retry...
|
2 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.
|
3 years ago |
|
jit.py
|
disable nvfuser for jit te/legacy modes (for PT 1.12+)
|
2 years ago |
|
log.py
|
Reorg of utils into separate modules
|
4 years ago |
|
metrics.py
|
Tweak accuracy topk safety. Fix #807
|
3 years ago |
|
misc.py
|
Reorg of utils into separate modules
|
4 years ago |
|
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
|
3 years ago |
|
summary.py
|
Make wandb optional
|
3 years ago |
