Variable number of objects

[MordehayM]

Hi,
First of all, great project!
I would like to ask about your dataset. it seems to me that each sample returned by the AudioVisualMUSICDataset's getitem method can be with variable number of objects(the "visuals" variable here:

co-separation/data/audioVisual_dataset.py

Line 189 in bd4f4fd

data['visuals'] = visuals

).
A question that arises from this is how the dataloader will batch these samples together.
I can see in the attached link that you address this problem by padding with -1 the lables(here:

co-separation/data/audioVisual_dataset.py

Line 183 in bd4f4fd

labels = np.vstack(objects_labels) #labels for each object, -1 denotes padded object

) but I did'nt figure it out where exactly you did this padding. Moreover, what about the variable objects problem mentioned above?

Thank you in advance