Jennnnnny commited on
Commit
ef7530e
1 Parent(s): 5e9f222

Update 修改后/4-5.Bert-seq2seq/train.py

Browse files
修改后/4-5.Bert-seq2seq/train.py CHANGED
@@ -11,7 +11,7 @@ N_EPOCHS = 30
11
  LR = 5e-4
12
  WARMUP_PROPORTION = 0.1
13
  MAX_GRAD_NORM = 1.0
14
- MODEL_PATH ='/Users/jenny/Desktop/修改后/4-5.Bert-seq2seq/bert-base-chinese'
15
  SAVE_PATH = './saved_models/pytorch_model.bin'
16
  device = "cuda" if torch.cuda.is_available() else 'cpu'
17
 
 
11
  LR = 5e-4
12
  WARMUP_PROPORTION = 0.1
13
  MAX_GRAD_NORM = 1.0
14
+ MODEL_PATH ='Jennnnnny/JennyLi/修改后/4-5.Bert-seq2seq/bert-base-chinese'
15
  SAVE_PATH = './saved_models/pytorch_model.bin'
16
  device = "cuda" if torch.cuda.is_available() else 'cpu'
17