ComfyUI Node: NNT TorchVision Data Loader
Category
NNT Neural Network Toolkit/Data Loading
Inputs
dataset_name
- CIFAR10
- CIFAR100
- MNIST
- FashionMNIST
- EMNIST
- SVHN
- STL10
- ImageNet
- LSUN
- CelebA
split
- train
- test
data_dir STRING
download
- True
- False
normalize_data
- True
- False
enable_augmentation
- True
- False
samples_to_return INT
start_index INT
use_cache
- True
- False
Outputs
TENSOR
TENSOR
STRING
INT
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