nroggendorff commited on
Commit
a2692eb
·
verified ·
1 Parent(s): dcc51de

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -134,7 +134,7 @@ def train_model(model, tokenizer, dataset, push, isinst):
134
  fp16=FP16,
135
  save_steps=int(1e+10),
136
  logging_steps=10,
137
- evaluation_strategy="steps",
138
  eval_steps=500,
139
  save_total_limit=2,
140
  )
 
134
  fp16=FP16,
135
  save_steps=int(1e+10),
136
  logging_steps=10,
137
+ evaluation_strategy="no",
138
  eval_steps=500,
139
  save_total_limit=2,
140
  )