I did consider this, but then would I still need another internal loop to run all the chunks through the model? In my head it looks like this…
for idx, batch in enumerate(dataloader):
     for x in y:
          code that either feeds chunks into the model 
          or code that takes chunks of the audio file and feeds it into model

Is there a way I could get the dataset to break it into chunks but also only return one chunk at a time?