develop #11
|
@ -51,7 +51,7 @@ dataset_params = {
|
||||||
trainer.load_data(dataset_params=dataset_params, batch_size=1)
|
trainer.load_data(dataset_params=dataset_params, batch_size=1)
|
||||||
|
|
||||||
# Fine-tune the model
|
# Fine-tune the model
|
||||||
trainer.finetune(output_path="trained_models")
|
trainer.finetune(output_path="trained_model")
|
||||||
```
|
```
|
||||||
|
|
||||||
## Dataset
|
## Dataset
|
||||||
|
|
Loading…
Reference in New Issue