Ryukijano commited on
Commit
f4e12b3
1 Parent(s): 06d442d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
README.md CHANGED
@@ -32,6 +32,16 @@ More information needed
32
  More information needed
33
 
34
  ## Training procedure
 
 
 
 
 
 
 
 
 
 
35
 
36
  ### Training hyperparameters
37
 
 
32
  More information needed
33
 
34
  ## Training procedure
35
+ from transformers import Seq2SeqTrainer
36
+
37
+ trainer = Seq2SeqTrainer(
38
+ args=training_args,
39
+ model=model,
40
+ train_dataset=dataset["train"],
41
+ eval_dataset=dataset["test"],
42
+ data_collator=data_collator,
43
+ tokenizer=processor,
44
+ )
45
 
46
  ### Training hyperparameters
47