minimind/dataset
xiao-baia c972c4e090
Fix DPO loss_mask boundary (include first assistant token)
Fix off-by-one in generate_loss_mask so the first assistant token contributes to loss in DPO.
2026-01-07 21:00:46 +08:00
..
__init__.py 250426 2025-04-26 10:07:55 +08:00
dataset.md [fix] git track 2025-10-21 23:09:00 +08:00
lm_dataset.py Fix DPO loss_mask boundary (include first assistant token) 2026-01-07 21:00:46 +08:00