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	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	Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.	2 years ago
jit.py	disable nvfuser for jit te/legacy modes (for PT 1.12+)	2 years ago
log.py	…
metrics.py	…
misc.py	…
model.py	Fix some formatting in utils/model.py	3 years ago
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