Solved by ptrblck in post #2 
              
              
                Based on the stack trace it seems you are using a custom Dataset, which seems to be defined in ./util/data_processing.py. Inside the __getitem__ method you are apparently using np.random.choice, which get’s an invalid argument combination. 
It seems that len(pt_img) might be smaller than self.pt_len …