small docs change

This commit is contained in:
Yinon Polak 2023-03-23 12:43:37 +02:00
parent c44b5b1b3a
commit 952e641213
1 changed files with 2 additions and 2 deletions

View File

@ -87,7 +87,7 @@ Mandatory parameters are marked as **Required** and have to be set in one of the
### PyTorch parameters
#### general
#### general:
| Parameter | Description |
|------------|-------------|
@ -96,7 +96,7 @@ Mandatory parameters are marked as **Required** and have to be set in one of the
| `model_kwargs` | paramters to be passed to the model class. <br> **Datatype:** dict. <br> Default: `{}`.
| `trainer_kwargs` | paramters to be passed to the trainer class. <br> **Datatype:** dict. <br> Default: `{}`.
#### trainer_kwargs
#### trainer_kwargs:
| Parameter | Description |
|------------|-------------|