Hugging Face's logo

woywan
/
t_n

Model card Files
xet
 Community
woywan commited on
Commit
5df8aab
·
verified ·
1 Parent(s): 4b5cf9f

Update pretrain.py

Browse files
Files changed (1) hide show
  1. pretrain.py +3 -3
pretrain.py CHANGED
@@ -19,9 +19,9 @@ NUM_LAYERS = 4
19
  DIM_FEEDFORWARD = 512
20
  DROPOUT = 0.1
21
  MAX_LEN = 4096
22
- SAVE_DIR = "d:/图像/novel_model"
23
- DATA_PATH = "d:/图像/novel_data/novel_dataset.pt"
24
- TOKENIZER_PATH = "d:/图像/novel_tokenizer.json"
25
 
26
  # 创建保存目录
27
  os.makedirs(SAVE_DIR, exist_ok=True)
 
19
  DIM_FEEDFORWARD = 512
20
  DROPOUT = 0.1
21
  MAX_LEN = 4096
22
+ SAVE_DIR = "./novel_model"
23
+ DATA_PATH = "./novel_data/novel_dataset.pt"
24
+ TOKENIZER_PATH = "./novel_tokenizer.json"
25
 
26
  # 创建保存目录
27
  os.makedirs(SAVE_DIR, exist_ok=True)