Added Ptn Pre Add Channel Axis node and Ptn Pre Flatten node. Also added example workflows to build a model from scratch using Ptn Chained Model node.
This commit is contained in:
@@ -120,6 +120,8 @@
|
||||
| [Ptn Linear](ptn_linear.md) | A linear model consisting of a single dense layer. |
|
||||
| [Ptn Linear Model](ptn_linear_model.md) | A linear model consisting of dense layers. |
|
||||
| [Ptn Max Pool](ptn_max_pool2d.md) | A maxpool layer. |
|
||||
| [Ptn Pre Add Channel Axis](ptn_pre_add_channel_axis.md) | Adds a channel axis after the batch axis if the input is rank 3 (bs, h, w) |
|
||||
| [Ptn Pre Flatten](ptn_pre_flatten.md) | Flattens the input tensor before processing the tensor in the specified model. |
|
||||
| [Ptn Resnet Model](ptn_resnet_model.md) | A Resnet model consisting of multiple Resnet layers. |
|
||||
## PyTorch wrapper - Optimizer
|
||||
| Node | Description |
|
||||
|
||||
@@ -25,7 +25,7 @@ t.
|
||||
| dilation | String |
|
||||
| groups | Int |
|
||||
| bias | Boolean |
|
||||
| padding_mode | Zeros |
|
||||
| padding_mode | |
|
||||
|
||||
## Output
|
||||
| Data type |
|
||||
|
||||
@@ -0,0 +1,18 @@
|
||||
# Ptn Pre Add Channel Axis
|
||||
Adds a channel axis after the batch axis if the input is rank 3 (bs, h, w)
|
||||
before the forward pass by the specified model.
|
||||
|
||||
## Input
|
||||
| Name | Data type |
|
||||
|---|---|
|
||||
| model | Ptmodel |
|
||||
|
||||
## Output
|
||||
| Data type |
|
||||
|---|
|
||||
| Ptmodel |
|
||||
|
||||
<HR>
|
||||
Category: PyTorch wrapper - Model
|
||||
|
||||
ComfyUI Pt Wrapper Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved.
|
||||
@@ -0,0 +1,17 @@
|
||||
# Ptn Pre Flatten
|
||||
Flattens the input tensor before processing the tensor in the specified model.
|
||||
|
||||
## Input
|
||||
| Name | Data type |
|
||||
|---|---|
|
||||
| model | Ptmodel |
|
||||
|
||||
## Output
|
||||
| Data type |
|
||||
|---|
|
||||
| Ptmodel |
|
||||
|
||||
<HR>
|
||||
Category: PyTorch wrapper - Model
|
||||
|
||||
ComfyUI Pt Wrapper Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved.
|
||||
Reference in New Issue
Block a user