Thank you for the reply!
Each .npy of input is less than 1GB and averages about 700MB.
The label has a smaller capacity.
The total train capacity combined with input and label is about 48GB.
There are a total of 58 input npy and label npy,
For each dataset, one input and one label are loaded.
A total of 58 datasets are created, and ConcatDataset is used to create one dataset (Final train dataset).