Skip to content

Commit babc07f

Browse files
author
Ronald Tse
committed
fix: labels.sha write belongs to distill_sequence only (logit-KD has no labels file)
1 parent 3ad0cc9 commit babc07f

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

‎src/gpu/modal_distill.py‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -399,7 +399,6 @@ def val_loss() -> float:
399399
if step % save_every == 0:
400400
ck = out_root / f"step-{step}"
401401
ck.mkdir(exist_ok=True)
402-
(ck / "labels.sha").write_text(labels_digest)
403402
torch.save(student.state_dict(), ck / "student.pt")
404403
torch.save(optimizer.state_dict(), ck / "optim.pt")
405404
CHECKPOINTS.commit()

0 commit comments

Comments
 (0)