ComfyUI Node: NNT Fine Tune Model
Category
NNT Neural Network Toolkit/Models
Inputs
MODEL MODEL
train_data TENSOR
train_labels TENSOR
val_data TENSOR
val_labels TENSOR
learning_rate FLOAT
epochs INT
batch_size INT
loss_function
- L1Loss
- MSELoss
- CrossEntropyLoss
- CTCLoss
- NLLLoss
- PoissonNLLLoss
- GaussianNLLLoss
- KLDivLoss
- BCELoss
- BCEWithLogitsLoss
- MarginRankingLoss
- HingeEmbeddingLoss
- MultiLabelMarginLoss
- HuberLoss
- SmoothL1Loss
- SoftMarginLoss
- MultiLabelSoftMarginLoss
- CosineEmbeddingLoss
- MultiMarginLoss
- TripletMarginLoss
- TripletMarginWithDistanceLoss
optimizer
- Adadelta
- Adagrad
- Adam
- AdamW
- SparseAdam
- Adamax
- ASGD
- NAdam
- RAdam
- RMSprop
- Rprop
- SGD
optimizer_params STRING
use_scheduler
- True
- False
scheduler
- StepLR
- ReduceLROnPlateau
- ExponentialLR
- CosineAnnealingLR
scheduler_params STRING
early_stopping
- True
- False
early_stopping_patience INT
save_best_model
- True
- False
best_model_path STRING
Outputs
MODEL
STRING
Extension: ComfyUI Neural Network Toolkit NNT
Neural Network Toolkit (NNT) for ComfyUI is an extensive set of custom ComfyUI nodes for designing, training, and fine-tuning neural networks. This toolkit allows defining models, layers, training workflows, transformers, and tensor operations in a visual manner using nodes.
Authored by inventorado
Run ComfyUI workflows in the Cloud!
No downloads or installs are required. Pay only for active GPU usage, not idle time. No complex setups and dependency issues
Learn more