Allow passing custom DataLoaders to SoundStreamTrainer

This allows customizing the data loading from the default provided
by SoundDataset, which is a folder of audio files on disk. For example,
it can allow someone to connect a WebDataset loader instead.

Also adds types to SoundStreamTrainer's constructor.