LH-Tech-AI
/

Excerp_v1

Question Answering

Model card Files Files and versions

LH-Tech-AI commited on 9 days ago

Commit

0b2f33e

·

verified ·

1 Parent(s): 78d6be8

Update train.py

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -24,7 +24,7 @@ DOC_STRIDE      = 128
 BATCH_SIZE      = 16
 EPOCHS          = 3
 LR              = 3e-4
-OUTPUT_DIR      = "distill"
 # ── 1. Dataset ───────────────────────────────────────────────
 raw = load_dataset("squad")

 BATCH_SIZE      = 16
 EPOCHS          = 3
 LR              = 3e-4
+OUTPUT_DIR      = "Excerp"
 # ── 1. Dataset ───────────────────────────────────────────────
 raw = load_dataset("squad")