diff --git a/README.md b/README.md index 2739809..849c45f 100644 --- a/README.md +++ b/README.md @@ -53,6 +53,8 @@ still integrate LogIX into their existing training code similarly to any traditi ### 🤗 HuggingFace Integration +A full example can be found [here](https://github.com/logix-project/logix/tree/main/examples/huggingface). + ```python from transformers import Trainer, Seq2SeqTrainer from logix.huggingface import patch_trainer, LogIXArguments @@ -82,6 +84,8 @@ trainer.self_influence() ### ⚡ PyTorch Lightning Integration +A full example can be found [here](https://github.com/logix-project/logix/tree/main/examples/lightning). + ```python from lightning import LightningModule, Trainer from logix.lightning import patch, LogIXArguments diff --git a/logix/config.py b/logix/config.py index 44f7396..3361100 100644 --- a/logix/config.py +++ b/logix/config.py @@ -57,7 +57,7 @@ class LoggingConfig: log_dir: str = field(init=False) flush_threshold: int = field( - default=-1, metadata={"help": "Flush threshold for the log buffer."} + default=1000000000, metadata={"help": "Flush threshold for the log buffer."} ) num_workers: int = field( default=1, metadata={"help": "Number of workers used for logging."}