nodes/HowToSD-ComfyUI-Pt-Wrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated model training document to add ResNet.

Browse Source
This commit is contained in:
HowToSD
2025-03-10 20:19:13 -07:00
parent 1dab39c78a
commit 2f52351247
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/model_training.md
+8 -3
View File
@@ -1,8 +1,13 @@
# Model Training
Experimental PyTorch model training has been added.
No coding is required. Training is as simple as a single click!
Currently, only classification is supported.
You can traing a PyTorch model without any coding.
Training is as simple as a single click!
Currently, classification models are supported for following architecture (node names are in parentheses):
* Linear (Ptn Linear Model)
* Convolutional Network (Ptn Conv Model)
* ResNet (Ptn Resnet Model)
You can adjust the model configuration including output dimension on the UI.
The system is designed to be modular. For example, to switch from a linear model to a convolutional model, you only need to replace the model node (see screenshot below).