ComfyUI Node
NNT TorchVision Datasets
A ComfyUI node in NNT Neural Network Toolkit/Tensors with 8 inputs and 4 outputs.
NNT TorchVision Datasets
- images
- labels
- dataset_info
- num_classes
◄dataset_nameCIFAR10►
◄splittrain►
◄data_dirdata/torchvision►
◄downloadTrue►
◄normalize_dataTrue►
◄enable_augmentationTrue►
◄samples_to_return32►
◄start_index0►
CategoryNNT Neural Network Toolkit/Tensors
Inputs (8)
| Name | Type | Default | Description |
|---|---|---|---|
| dataset_name | COMBO | CIFAR10 | 4 options: CIFAR10, CIFAR100, MNIST, FashionMNIST |
| split | COMBO | train | 2 options: train, test |
| data_dir | STRING | data/torchvision | — |
| download | COMBO | True | 2 options: True, False |
| normalize_data | COMBO | True | 2 options: True, False |
| enable_augmentation | COMBO | True | 2 options: True, False |
| samples_to_return | INT | 321–512 | — |
| start_index | INT | 00–50000 | — |
Outputs (4)
| Name | Type | Description |
|---|---|---|
| images | TENSOR | — |
| labels | TENSOR | — |
| dataset_info | STRING | — |
| num_classes | INT | — |