ComfyUI Node
NNT Edit Model Layers
A ComfyUI node in NNT Neural Network Toolkit/Models with 10 inputs and 3 outputs.
NNT Edit Model Layers
- MODEL
- edited_model
- info_message
- layer_stats
◄operationFreeze►
◄parameter_typeboth►
◄layer_selectionAll layers►
◄layer_typesAll►
◄num_layers1►
◄initializationkaiming_normal►
◄custom_value0.0►
◄pruning_amount0.5►
◄quantization_bits8►
CategoryNNT Neural Network Toolkit/Models
Inputs (10)
| Name | Type | Default | Description |
|---|---|---|---|
| MODEL | MODEL | — | |
| operation | COMBO | Freeze | 5 options: Freeze, Unfreeze, Set weights and biases, Prune, Quantize |
| parameter_type | COMBO | both | 3 options: weights, biases, both |
| layer_selection | COMBO | All layers | 4 options: All layers, First N layers, Last N layers, Selected types |
| layer_types | COMBO | All | 4 options: Linear, Conv2d, BatchNorm2d, All |
| num_layers | INT | 11–1000 | — |
| initialization | COMBO | kaiming_normal | 9 options: random, zeros, ones, xavier_uniform, xavier_normal, kaiming_uniform, +3 |
| custom_value | FLOAT | 0.0-10000000000–10000000000 | — |
| pruning_amount | FLOAT | 0.50–0.9 | — |
| quantization_bits | INT | 84–32 | — |
Outputs (3)
| Name | Type | Description |
|---|---|---|
| edited_model | MODEL | — |
| info_message | STRING | — |
| layer_stats | DICT | — |