ComfyUI Node
GridSearchTraining
A ComfyUI node in ETK/pytorch with 5 inputs and 3 outputs.
GridSearchTraining
- model
- dataset
- features tensor
- labels tensor
- best_model
- best_params
- all_metrics
◄param_grid{"epochs": [1, 2], "batch_size": [32, 64], "loss_function": ["MSELoss", "CrossEntropyLoss"]}►
CategoryETK/pytorch
Inputs (5)
| Name | Type | Default | Description |
|---|---|---|---|
| model | TORCH_MODEL | — | |
| param_grid | STRING | {"epochs": [1, 2], "batch_size": [32, 64], "loss_function": ["MSELoss", "CrossEntropyLoss"]} | — |
| datasetopt | TORCH_DATASET | — | |
| features tensoropt | TORCH_TENSOR | — | |
| labels tensoropt | TORCH_TENSOR | — |
Outputs (3)
| Name | Type | Description |
|---|---|---|
| best_model | TORCH_MODEL | — |
| best_params | DICT | — |
| all_metrics | LIST | — |