small docs change

This commit is contained in:
Yinon Polak 2023-03-23 12:43:37 +02:00
parent c44b5b1b3a
commit 952e641213

View File

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