Update train.py

train.py

@@ -7,7 +7,7 @@ from transformers import (
 )
 import zipfile
 
-ROOT = Path(_file_).resolve().parent
+ROOT = Path(__file__).resolve().parent
 
 def parse_args():
     ap = argparse.ArgumentParser()