Update README.md
Browse files
README.md
CHANGED
|
@@ -45,7 +45,7 @@ You can use the following code template to load and fine-tune the model
|
|
| 45 |
compute_metrics=compute_metrics
|
| 46 |
)
|
| 47 |
|
| 48 |
-
# train the
|
| 49 |
trainer.train()
|
| 50 |
predictions = trainer.predict(eval_split)
|
| 51 |
with open(f"{output_dir}predictions.pickle", "wb") as fp:
|
|
|
|
| 45 |
compute_metrics=compute_metrics
|
| 46 |
)
|
| 47 |
|
| 48 |
+
# train the model
|
| 49 |
trainer.train()
|
| 50 |
predictions = trainer.predict(eval_split)
|
| 51 |
with open(f"{output_dir}predictions.pickle", "wb") as fp:
|