This commit is contained in:
Jeff Omhover 2022-09-06 22:04:57 -07:00
Родитель a083781f26
Коммит cb059f57a9
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -219,7 +219,7 @@ def main():
tokenizer=tokenizer,
data_collator=data_collator,
compute_metrics=lambda preds : compute_metrics(preds, tokenizer, metric),
callbacks=[CustomMetricCallback]
callbacks=[CustomCallback]
)
# Start the actual training (to include evaluation use --do-eval)