ComfyUI Node

TensorsToDataset

A ComfyUI node in ETK/pytorch with 3 inputs and 1 output.

By TashaSkyUp·Created about a year ago·Updated about a year ago· 1
TensorsToDataset
  • input_data
  • labels
  • TORCH_DATASET
move to devicecpu
CategoryETK/pytorch

Inputs (3)

NameTypeDefaultDescription
input_dataTORCH_TENSOR
labelsTORCH_TENSOR
move to deviceoptCOMBOcpu2 options: cuda, cpu

Outputs (1)

NameTypeDescription
TORCH_DATASETTORCH_DATASET