Solved by ptrblck in post #2 
              
              
                The length of a Dataset is defined in it’s __len__() method and you could override it with a custom method returning your desired length or alternatively implement a new custom Dataset with the desired length.