diff --git a/README.md b/README.md index 80441e0..aeb0837 100644 --- a/README.md +++ b/README.md @@ -51,7 +51,7 @@ dataset_params = { trainer.load_data(dataset_params=dataset_params, batch_size=1) # Fine-tune the model -trainer.finetune(output_path="trained_models") +trainer.finetune(output_path="trained_model") ``` ## Dataset