ComfyUI Node
Pt Train RNN Model
A ComfyUI node in Training with 15 inputs and 3 outputs.
Pt Train RNN Model
- model
- train_loader
- optimizer
- loss_function
- scheduler
- val_loader
- h_0
- Model
- train loss
- val loss
◄linear_headtrue►
◄epochs1►
◄use_gpufalse►
◄early_stoppingfalse►
◄early_stopping_rounds10►
◄output_best_val_modeltrue►
◄classification_metricsfalse►
◄use_valid_token_meantrue►
CategoryTraining
Inputs (15)
| Name | Type | Default | Description |
|---|---|---|---|
| model | PTMODEL | — | |
| linear_head | BOOLEAN | true | — |
| train_loader | PTDATALOADER | — | |
| optimizer | PTOPTIMIZER | — | |
| loss_function | PTLOSS | — | |
| 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 | false | — |
| use_valid_token_mean | BOOLEAN | true | — |
| scheduleropt | PTLRSCHEDULER | — | |
| val_loaderopt | PTDATALOADER | — | |
| h_0opt | TENSOR | — |
Outputs (3)
| Name | Type | Description |
|---|---|---|
| Model | PTMODEL | — |
| train loss | TENSOR | — |
| val loss | TENSOR | — |