ComfyUI Node: Pt Train RNN 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
linear_head BOOLEAN
train_loader PTDATALOADER
optimizer PTOPTIMIZER
loss_function PTLOSS
epochs INT
use_gpu BOOLEAN
early_stopping BOOLEAN
early_stopping_rounds INT
output_best_val_model BOOLEAN
classification_metrics BOOLEAN
use_valid_token_mean BOOLEAN
scheduler PTLRSCHEDULER
val_loader PTDATALOADER
h_0 TENSOR
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.