Update README.md

sileod · Jun 8, 2024 · 9c1280a · 9c1280a
1 parent 2e53ce8
commit 9c1280a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/README.md b/README.md
@@ -32,8 +32,7 @@ class hparams:
   learning_rate = 3e-5 # see hf.co/docs/transformers/en/main_classes/trainer#transformers.TrainingArguments
 
 tasks = [rte]
-model = tn.Model(tasks, hparams)
-trainer = tn.Trainer(model, tasks, hparams)
+model, trainer = tn.Model_Trainer(task,hparams)
 trainer.train()
 trainer.evaluate()
 p = trainer.pipeline()