qwerrwe / src /axolotl /utils /callbacks.py

Commit History

mypy wandb ignore (#572)
c6d870b
unverified

winglian commited on

fix wandb so mypy doesn't complain (#562)
bf08044
unverified

winglian commited on

Add training callback to send predictions to WandB table (#521)
5b67ea9
unverified

Glavin001 commited on

Early stopping metric (#537)
e30f1e3
unverified

winglian commited on

No gather single gpu (#523)
09f1543
unverified

winglian commited on

Changed Bench Eval to report metrics correctly by split. Added total accuracy and renamed previously used bench_accuracy to bench_average_accuracy. (#512)
42f9642
unverified

Alpay Ariyak commited on

add eval benchmark callback (#441)
7657632
unverified

winglian commited on

ReLoRA implementation (with quantization) (#322)
bde3c5a
unverified

chargoddard winglian commited on

improve GPU logging to break out pytorch cache and system mem
7b55fe6

tmm1 commited on

log GPU memory usage
e303d64

tmm1 commited on

more gpt-neox long ctx fixes
ab5cd28

winglian commited on

fix bettertransformers save, force it to skip after saving correctly in callback
1a82082

winglian commited on

more tweaks to do pre-training with bettertransformers
1210dc8

winglian commited on

Apply isort then black
37293dc

Nanobit commited on

Lint callbacks.py
5062eca

Nanobit commited on

black formatting
2bc1a5b

winglian commited on

Add callback save peft_model on_save
0d6708b

Nanobit commited on