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 b1b024dfed Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168		2 years ago
..
__init__.py	Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.	2 years ago
agc.py	…
checkpoint_saver.py	…
clip_grad.py	…
cuda.py	…
decay_batch.py	…
distributed.py	Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.	2 years ago
jit.py	…
log.py	…
metrics.py	…
misc.py	…
model.py	…
model_ema.py	…
random.py	…
summary.py	Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168	2 years ago