25 lines
510 B
Markdown
25 lines
510 B
Markdown
# Pt Train Classification Model
|
|
Trains a classification model using a given dataset, optimizer, and number of epochs.
|
|
|
|
## Input
|
|
| Name | Data type |
|
|
|---|---|
|
|
| model | Ptmodel |
|
|
| train_loader | Ptdataloader |
|
|
| optimizer | Ptoptimizer |
|
|
| epochs | Int |
|
|
| use_gpu | Boolean |
|
|
| val_loader | Ptdataloader |
|
|
|
|
## Output
|
|
| Data type |
|
|
|---|
|
|
| Ptmodel |
|
|
| Tensor |
|
|
| Tensor |
|
|
|
|
<HR>
|
|
Category: PyTorch wrapper - Training
|
|
|
|
ComfyUI Data Analysis Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved.
|