加载训练数据集 - ComfyUI Built-in Node Documentation

此节点加载一个之前已保存到磁盘的编码训练数据集（latents 和 conditioning）。

此节点加载一个之前已保存到磁盘的编码训练数据集（latents 和 conditioning）。它会从 datasets 目录中所选的数据集文件夹读取所有 shard_*.pkl 数据分片文件，并返回合并后的 latent 向量和 conditioning 数据，供训练工作流使用。

输入

参数	描述	数据类型	必需	范围
`文件夹名`	要从 datasets 目录加载的已保存数据集。	COMBO	是	datasets 目录中找到的每个数据集文件夹对应一个选项

注意：folder_name 选项通过扫描 datasets 目录自动构建。当子文件夹包含 metadata.json 文件或至少一个 .safetensors 文件时，该子文件夹会被列为数据集（扫描不会进入匹配的文件夹内部）。所选数据集文件夹会在所有已配置的数据集根目录中搜索，并且该文件夹名称必须解析为其中一个根目录内的子文件夹。此节点按排序顺序读取所选文件夹中所有名为 shard_*.pkl 的文件，如果未找到分片文件或无法定位该文件夹，则会引发错误。

输出

输出名称	描述	数据类型
`latents`	latent 字典列表（输出列表），其中每个字典包含一个带张量的 `"samples"` 键。	LATENT
`conditioning`	conditioning 列表的列表（输出列表），其中每个内部列表包含对应样本的 conditioning 数据。	CONDITIONING

本文档由 AI 生成。如果您发现任何错误或有改进建议，欢迎贡献！在 GitHub 上编辑

Source fingerprint (SHA-256): 9f914b27f067460f6f3b54f3f2a7bb793c65b99c85e8aa14ab64894be26bd816