qwerrwe / src /axolotl /utils /callbacks.py

Commit History

[WandB] Push axolotl config to top level wandb files (#1014)
4f4d638
unverified

hamel commited on

feature: loss watchdog for terminating training runs that are failing (#899)
58ec8b1
unverified

user735 Karl-Johan Alm commited on

refactor setup trainer so we can add more hooks (#773)
6c81c61
unverified

winglian commited on

Save Axolotl config as WandB artifact (#716)
490923f
unverified

Jan Philipp Harries commited on

chore(callback): Remove old peft saving code (#510)
d5f8589
unverified

Nanobit commited on

run eval on the first step to get a baseline (#617)
2844eb2
unverified

winglian commited on

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