ComfyUI Node
Pt Train Fine Tune Classification Transformer Model
A ComfyUI node in Training with 12 inputs and 3 outputs.
Pt Train Fine Tune Classification Transformer Model
- model
- train_loader
- optimizer
- loss_function
- val_loader
- Model
- train loss
- val loss
◄epochs2►
◄freeze_pretrained_module_epochs1►
◄use_gpufalse►
◄early_stoppingfalse►
◄early_stopping_rounds10►
◄output_best_val_modeltrue►
◄classification_metricstrue►
CategoryTraining
Inputs (12)
| Name | Type | Default | Description |
|---|---|---|---|
| model | PTMODEL | — | |
| train_loader | PTDATALOADER | — | |
| optimizer | PTOPTIMIZER | — | |
| loss_function | PTLOSS | — | |
| epochs | INT | 21–1000000 | — |
| freeze_pretrained_module_epochs | INT | 11–1000000 | — |
| use_gpu | BOOLEAN | false | — |
| early_stopping | BOOLEAN | false | — |
| early_stopping_rounds | INT | 101–1000 | — |
| output_best_val_model | BOOLEAN | true | — |
| classification_metrics | BOOLEAN | true | — |
| val_loaderopt | PTDATALOADER | — |
Outputs (3)
| Name | Type | Description |
|---|---|---|
| Model | PTMODEL | — |
| train loss | TENSOR | — |
| val loss | TENSOR | — |