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
dff33730b3
Merge remote-tracking branch 'origin/master' into bits_and_tpu
|
3 years ago |
| .. |
|
__init__.py
|
Add --fuser arg to train/validate/benchmark scripts to select jit fuser type
|
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
|
Disable nvfuser fma / opt level overrides per #1244
|
3 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
|
4 years ago |
|
summary.py
|
Make wandb optional
|
4 years ago |
