Spaces:
Sleeping
Sleeping
Update train.py
Browse files
train.py
CHANGED
|
@@ -6,7 +6,7 @@ from transformers import (
|
|
| 6 |
DataCollatorForLanguageModeling, Trainer, TrainingArguments
|
| 7 |
)
|
| 8 |
|
| 9 |
-
ROOT = Path(
|
| 10 |
DONE = ROOT / "TRAIN_DONE" # <- write here
|
| 11 |
ERRF = ROOT / "TRAIN_ERROR"
|
| 12 |
|
|
|
|
| 6 |
DataCollatorForLanguageModeling, Trainer, TrainingArguments
|
| 7 |
)
|
| 8 |
|
| 9 |
+
ROOT = Path(__file__).resolve().parent # /home/user/app
|
| 10 |
DONE = ROOT / "TRAIN_DONE" # <- write here
|
| 11 |
ERRF = ROOT / "TRAIN_ERROR"
|
| 12 |
|