File tree Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -53,6 +53,8 @@ still integrate LogIX into their existing training code similarly to any traditi
53
53
54
54
### 🤗 HuggingFace Integration
55
55
56
+ A full example can be found [ here] ( https://github.com/logix-project/logix/tree/main/examples/huggingface ) .
57
+
56
58
``` python
57
59
from transformers import Trainer, Seq2SeqTrainer
58
60
from logix.huggingface import patch_trainer, LogIXArguments
@@ -82,6 +84,8 @@ trainer.self_influence()
82
84
83
85
### ⚡ PyTorch Lightning Integration
84
86
87
+ A full example can be found [ here] ( https://github.com/logix-project/logix/tree/main/examples/lightning ) .
88
+
85
89
``` python
86
90
from lightning import LightningModule, Trainer
87
91
from logix.lightning import patch, LogIXArguments
Original file line number Diff line number Diff line change @@ -57,7 +57,7 @@ class LoggingConfig:
57
57
58
58
log_dir : str = field (init = False )
59
59
flush_threshold : int = field (
60
- default = - 1 , metadata = {"help" : "Flush threshold for the log buffer." }
60
+ default = 1000000000 , metadata = {"help" : "Flush threshold for the log buffer." }
61
61
)
62
62
num_workers : int = field (
63
63
default = 1 , metadata = {"help" : "Number of workers used for logging." }
You can’t perform that action at this time.
0 commit comments