ComfyUI Node: Pt Train Regression Model
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
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
Looking for a different node?
More nodes in ComfyUI-Pt-Wrapper
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.