ComfyUI Node
NNT TorchVision Data Loader
A ComfyUI node in NNT Neural Network Toolkit/Data Loading with 9 inputs and 4 outputs.
NNT TorchVision Data Loader
- images
- labels
- dataset_info
- num_classes
◄dataset_nameMNIST►
◄splittrain►
◄data_dir►
◄downloadTrue►
◄normalize_dataTrue►
◄enable_augmentationFalse►
◄samples_to_return32►
◄start_index0►
◄use_cacheTrue►
CategoryNNT Neural Network Toolkit/Data Loading
Inputs (9)
| Name | Type | Default | Description |
|---|---|---|---|
| dataset_name | COMBO | MNIST | 10 options: CIFAR10, CIFAR100, MNIST, FashionMNIST, EMNIST, SVHN, +4 |
| split | COMBO | train | 2 options: train, test |
| data_dir | STRING | — | |
| download | COMBO | True | 2 options: True, False |
| normalize_data | COMBO | True | 2 options: True, False |
| enable_augmentation | COMBO | False | 2 options: True, False |
| samples_to_return | INT | 321–50000 | — |
| start_index | INT | 00–50000 | — |
| use_cache | COMBO | True | 2 options: True, False |
Outputs (4)
| Name | Type | Description |
|---|---|---|
| images | TENSOR | — |
| labels | TENSOR | — |
| dataset_info | STRING | — |
| num_classes | INT | — |