How to use pytroch to load different size datasets?

Question

I have written a siamese network and want to train it.

The code I have written:

txt_path = \'home\\xxx\\train.txt\'  #contains all different npy files.