ComfyUI Node: Pt Train Regression Model
Category
Training
Inputs
model PTMODEL
train_loader PTDATALOADER
optimizer PTOPTIMIZER
epochs INT
use_gpu BOOLEAN
early_stopping BOOLEAN
early_stopping_rounds INT
output_best_val_model BOOLEAN
val_loader PTDATALOADER
Outputs
PTMODEL
TENSOR
TENSOR
Extension: ComfyUI-Pt-Wrapper
PyTorch extension for ComfyUI featuring extensive PyTorch wrapper nodes for seamless tensor operations and PyTorch model training.
Authored by HowToSD
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