ComfyUI > Nodes > ComfyUI > Load Training Dataset

ComfyUI Node: Load Training Dataset

Class Name

LoadTrainingDataset

Category
dataset
Author
ComfyAnonymous (Account age: 763days)
Extension
ComfyUI
Latest Updated
2026-05-13
Github Stars
112.77K

How to Install ComfyUI

Install this extension via the ComfyUI Manager by searching for ComfyUI
  • 1. Click the Manager button in the main menu
  • 2. Select Custom Nodes Manager button
  • 3. Enter ComfyUI in the search bar
After installation, click the Restart button to restart ComfyUI. Then, manually refresh your browser to clear the cache and access the updated list of nodes.

Visit ComfyUI Online for ready-to-use ComfyUI environment

  • Free trial available
  • 16GB VRAM to 80GB VRAM GPU machines
  • 400+ preloaded models/nodes
  • Freedom to upload custom models/nodes
  • 200+ ready-to-run workflows
  • 100% private workspace with up to 200GB storage
  • Dedicated Support

Run ComfyUI Online

Load Training Dataset Description

Facilitates importing encoded training datasets for AI model training, streamlining workflow and saving time.

Load Training Dataset:

The LoadTrainingDataset node is designed to facilitate the process of importing encoded training datasets from disk into your workflow. This node is particularly useful for AI artists and developers who need to work with pre-processed datasets for training machine learning models. By loading datasets that have been previously encoded and saved, you can streamline your workflow, saving time and computational resources. The node is experimental, indicating that it may be subject to changes or improvements. Its primary function is to retrieve and prepare the necessary data components, such as latent representations and conditioning information, which are essential for training tasks. This capability is crucial for those looking to efficiently manage and utilize large datasets in their AI projects.

Load Training Dataset Input Parameters:

folder_name

The folder_name parameter specifies the name of the folder containing the saved dataset within the output directory. This parameter is crucial as it directs the node to the correct location where the dataset is stored. The default value is "training_dataset", but you can change it to match the specific folder name where your dataset resides. This parameter does not have a minimum or maximum value, as it is a string input. Ensuring the correct folder name is provided is essential for the successful loading of the dataset.

Load Training Dataset Output Parameters:

latents

The latents output provides a list of latent dictionaries. These latents are encoded representations of your dataset, which are used as inputs for training machine learning models. They are crucial for the model to learn and generalize from the data. The output is a list, allowing for multiple latent representations to be handled simultaneously, which is beneficial for batch processing and parallel computations.

conditioning

The conditioning output consists of a list of conditioning lists. Conditioning information is used to guide the model during training, providing additional context or constraints that can influence the learning process. This output is essential for tasks that require specific conditions or parameters to be met during model training, ensuring that the model adapts to the desired specifications.

Load Training Dataset Usage Tips:

  • Ensure that the folder_name parameter accurately reflects the directory where your dataset is stored to avoid loading errors.
  • Utilize the latents and conditioning outputs to feed directly into your model training pipeline, ensuring that your model receives the necessary data and context for effective learning.
  • Consider organizing your datasets in a consistent manner within the output directory to streamline the loading process and reduce the likelihood of errors.

Load Training Dataset Common Errors and Solutions:

Dataset not found

  • Explanation: This error occurs when the specified folder_name does not exist in the output directory.
  • Solution: Verify that the folder name is correct and that the dataset is indeed located in the specified directory. Double-check for any typos or discrepancies in the folder name.

Incomplete dataset

  • Explanation: This error may arise if the dataset is missing some of the expected files or components.
  • Solution: Ensure that all necessary files are present in the dataset folder. If the dataset was manually moved or copied, verify that the transfer was complete and that no files were omitted.

Corrupted data files

  • Explanation: This error indicates that one or more files in the dataset are corrupted and cannot be read.
  • Solution: Check the integrity of the dataset files. If possible, re-encode or re-download the dataset to ensure that all files are intact and readable.

Load Training Dataset Related Nodes

Go back to the extension to check out more related nodes.
ComfyUI
RunComfy
Copyright 2025 RunComfy. All Rights Reserved.

RunComfy is the premier ComfyUI platform, offering ComfyUI online environment and services, along with ComfyUI workflows featuring stunning visuals. RunComfy also provides AI Models, enabling artists to harness the latest AI tools to create incredible art.

Load Training Dataset