ComfyUI Node: NNT Define Conv Layer
Category
NNT Neural Network Toolkit/Layers
Inputs
conv_type
- Conv1d
- Conv2d
- Conv3d
- ConvTranspose1d
- ConvTranspose2d
- ConvTranspose3d
- Unfold
- Fold
out_channels INT
kernel_size INT
stride INT
padding INT
padding_mode
- zeros
- reflect
- replicate
- circular
output_padding INT
dilation INT
groups INT
use_bias
- True
- False
activation_function
- None
- ELU
- GELU
- GLU
- Hardshrink
- Hardsigmoid
- Hardswish
- Hardtanh
- LeakyReLU
- LogSigmoid
- MultiheadAttention
- PReLU
- ReLU
- ReLU6
- RReLU
- SELU
- CELU
- Sigmoid
- SiLU
- Softmax
- Softmax2d
- Softmin
- Softplus
- Softshrink
- Softsign
- Tanh
- Tanhshrink
- Threshold
normalization
- None
- BatchNorm1d
- BatchNorm2d
- BatchNorm3d
- LayerNorm
- InstanceNorm1d
- InstanceNorm2d
- InstanceNorm3d
- GroupNorm
- LocalResponseNorm
norm_eps FLOAT
norm_momentum FLOAT
norm_affine
- True
- False
dropout_rate FLOAT
weight_init
- default
- normal
- uniform
- xavier_normal
- xavier_uniform
- kaiming_normal
- kaiming_uniform
- orthogonal
- sparse
- dirac
- zeros
- ones
weight_init_gain FLOAT
weight_init_mode
- fan_in
- fan_out
weight_init_nonlinearity
- relu
- leaky_relu
- selu
- tanh
- linear
- sigmoid
num_copies INT
LAYER_STACK LIST
hyperparameters DICT
Outputs
LIST
Extension: ComfyUI Neural Network Toolkit NNT
Neural Network Toolkit (NNT) for ComfyUI is an extensive set of custom ComfyUI nodes for designing, training, and fine-tuning neural networks. This toolkit allows defining models, layers, training workflows, transformers, and tensor operations in a visual manner using nodes.
Authored by inventorado
Run ComfyUI workflows in the Cloud!
No downloads or installs are required. Pay only for active GPU usage, not idle time. No complex setups and dependency issues
Learn more