(self, config, mesh, learning_rate_schedule, goodput_recorder)
| 46 | """Training hooks for SFT.""" |
| 47 | |
| 48 | def __init__(self, config, mesh, learning_rate_schedule, goodput_recorder): |
| 49 | self.config = config |
| 50 | self.mesh = mesh |
| 51 | self.metric_logger = MetricLogger(self.config, learning_rate_schedule) |
| 52 | self.goodput_recorder = goodput_recorder |
| 53 | self.metadata = {} |
| 54 | self.train_metadata = defaultdict(float) |
| 55 | self.eval_metadata = defaultdict(float) |
| 56 | |
| 57 | @override |
| 58 | def on_train_start(self, train_ctx: peft_trainer.PeftTrainer): |
nothing calls this directly
no test coverage detected