Thanks for the idea!
The idea you mentioned is not using ConcatDataset, but creating a single traindataset for each npy.
After that, train 58 traindatasets in the train loop as one epoch.
Is what I understand correct?
Thanks again for your help