Inputs (11)
| Name | Type | Required |
|---|---|---|
| model | TORCH_MODEL | required |
| dataset | TORCH_DATASET | optional |
| features tensor | TORCH_TENSOR | optional |
| labels tensor | TORCH_TENSOR | optional |
| epochs | INT | optional |
| batch_size | INT | optional |
| loss_function | COMBO | optional |
| optimizer | COMBO | optional |
| metrics | COMBO | optional |
| device | STRING | optional |
| create_samples | COMBO | optional |
Outputs (3)
| Name | Type |
|---|---|
| model | TORCH_MODEL |
| metrics | LIST |
| best model | TORCH_MODEL |