src.core.callbacks

callbacks.py

Custom Hugging Face Weights and Biases Callback that allows for writing custom metrics, better resuming functionality.

Functions

rewrite_logs

Classes

CustomCheckpointCallback	Custom Checkpoint Callback used by Mistral for Saving Checkpoints at different frequencies.
CustomWandbCallback	Custom Weights and Biases Callback used by Mistral for logging information from the Huggingface Trainer.