ComfyUI Node: TrainModel
Run ComfyUI workflows without the setup
No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.
Category
ETK/pytorch
Inputs
model TORCH_MODEL
dataset TORCH_DATASET
features tensor TORCH_TENSOR
labels tensor TORCH_TENSOR
epochs INT
batch_size INT
loss_function
- AdaptiveLogSoftmaxWithLoss
- BCELoss
- BCEWithLogitsLoss
- CTCLoss
- CosineEmbeddingLoss
- CrossEntropyLoss
- GaussianNLLLoss
- HingeEmbeddingLoss
- HuberLoss
- KLDivLoss
- L1Loss
- MSELoss
- MarginRankingLoss
- MultiLabelMarginLoss
- MultiLabelSoftMarginLoss
- MultiMarginLoss
- NLLLoss
- PoissonNLLLoss
- SmoothL1Loss
- SoftMarginLoss
- TripletMarginLoss
- TripletMarginWithDistanceLoss
optimizer
- ASGD
- Adadelta
- Adafactor
- Adagrad
- Adam
- AdamW
- Adamax
- LBFGS
- NAdam
- Optimizer
- RAdam
- RMSprop
- Rprop
- SGD
- SparseAdam
metrics
- accuracy
- AdaptiveLogSoftmaxWithLoss
- BCELoss
- BCEWithLogitsLoss
- CTCLoss
- CosineEmbeddingLoss
- CrossEntropyLoss
- GaussianNLLLoss
- HingeEmbeddingLoss
- HuberLoss
- KLDivLoss
- L1Loss
- MSELoss
- MarginRankingLoss
- MultiLabelMarginLoss
- MultiLabelSoftMarginLoss
- MultiMarginLoss
- NLLLoss
- PoissonNLLLoss
- SmoothL1Loss
- SoftMarginLoss
- TripletMarginLoss
- TripletMarginWithDistanceLoss
device STRING
create_samples
- TRUE
- FALSE
Outputs
TORCH_MODEL
LIST
TORCH_MODEL
Extension: EternalKernel PyTorch Nodes
Comprehensive PyTorch nodes for ComfyUI - Neural network training, inference, and ML workflows
Authored by TashaSkyUp
Looking for a different node?
More nodes in EternalKernel PyTorch Nodes
Run ComfyUI workflows without the setup
No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.