AssertionError in training

Hello, I ran into this problem when training the aero model:
assert len(self.hr_set) == len(self.lr_set) AssertionError

I should be caused by segmenting, can I set segment=None?

It's been a while since I looked at the code, but if I recall correctly, setting segment=None will cause the dataset to get each sample as a whole. You are welcome to dive into the code here:

aero/src/data/datasets.py

Line 130 in 63fc8a2

    
           self.lr_set = Audioset(lr, sample_rate=lr_sr, length=lr_length, stride=lr_stride, pad=pad, channels=1,

.

If you need more assistance, please share the config files you used, and print out the actual length of the hr_set and lr_set you are receiving.

Have you ever trained the model on the condition of 2-16? If you have, can you provide the pre-trained models?