ComfyUI Node
NNT Fine Tune Model
A ComfyUI node in NNT Neural Network Toolkit/Models with 18 inputs and 2 outputs.
NNT Fine Tune Model
- MODEL
- train_data
- train_labels
- val_data
- val_labels
- fine_tuned_model
- training_log
◄learning_rate0.0001►
◄epochs10►
◄batch_size32►
◄loss_functionCrossEntropyLoss►
◄optimizerAdam►
◄optimizer_params{}►
◄use_schedulerFalse►
◄schedulerStepLR►
◄scheduler_params{}►
◄early_stoppingFalse►
◄early_stopping_patience5►
◄save_best_modelTrue►
◄best_model_pathbest_model.pth►
CategoryNNT Neural Network Toolkit/Models
Inputs (18)
| Name | Type | Default | Description |
|---|---|---|---|
| MODEL | MODEL | — | |
| train_data | TENSOR | — | |
| train_labels | TENSOR | — | |
| val_data | TENSOR | — | |
| val_labels | TENSOR | — | |
| learning_rate | FLOAT | 0.00011e-8–1 | — |
| epochs | INT | 101–1000 | — |
| batch_size | INT | 321–1024 | — |
| loss_function | COMBO | CrossEntropyLoss | 21 options: L1Loss, MSELoss, CrossEntropyLoss, CTCLoss, NLLLoss, PoissonNLLLoss, +15 |
| optimizer | COMBO | Adam | 12 options: Adadelta, Adagrad, Adam, AdamW, SparseAdam, Adamax, +6 |
| optimizer_params | STRING | {} | — |
| use_scheduler | COMBO | False | 2 options: True, False |
| scheduler | COMBO | StepLR | 4 options: StepLR, ReduceLROnPlateau, ExponentialLR, CosineAnnealingLR |
| scheduler_params | STRING | {} | — |
| early_stopping | COMBO | False | 2 options: True, False |
| early_stopping_patience | INT | 51–100 | — |
| save_best_model | COMBO | True | 2 options: True, False |
| best_model_path | STRING | best_model.pth | — |
Outputs (2)
| Name | Type | Description |
|---|---|---|
| fine_tuned_model | MODEL | — |
| training_log | STRING | — |