small docs change

pull/8297/head
Yinon Polak 3 years ago
parent c44b5b1b3a
commit 952e641213

@ -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 |
|------------|-------------|

Loading…
Cancel
Save