Percy3822 commited on
Commit
6f02285
·
verified ·
1 Parent(s): 024546d

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -6,7 +6,7 @@ from transformers import (
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
 
 
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