ComfyUI Node
Pt Train Classification Model Lr
A ComfyUI node in Training with 10 inputs and 3 outputs.
Pt Train Classification Model Lr
- model
- train_loader
- optimizer
- scheduler
- val_loader
- Model
- train loss
- val loss
◄epochs1►
◄use_gpufalse►
◄early_stoppingfalse►
◄early_stopping_rounds10►
◄output_best_val_modeltrue►
CategoryTraining
Inputs (10)
| Name | Type | Default | Description |
|---|---|---|---|
| model | PTMODEL | — | |
| train_loader | PTDATALOADER | — | |
| optimizer | PTOPTIMIZER | — | |
| scheduler | PTLRSCHEDULER | — | |
| epochs | INT | 11–1000000 | — |
| use_gpu | BOOLEAN | false | — |
| early_stopping | BOOLEAN | false | — |
| early_stopping_rounds | INT | 101–1000 | — |
| output_best_val_model | BOOLEAN | true | — |
| val_loaderopt | PTDATALOADER | — |
Outputs (3)
| Name | Type | Description |
|---|---|---|
| Model | PTMODEL | — |
| train loss | TENSOR | — |
| val loss | TENSOR | — |