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
c2f02b08b8
Merge remote-tracking branch 'origin/attn_update' into bits_and_tpu
|
3 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
|
Tweak accuracy topk safety. Fix #807
|
3 years ago |
|
misc.py
|
Reorg of utils into separate modules
|
4 years ago |
|
model.py
|
Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go...
|
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 |
