Class limit for datamodules in the multi-label setting

DBD-research-group / BirdSet

A benchmark dataset collection for bird sound classification

https://huggingface.co/datasets/DBD-research-group/BirdSet

BSD 3-Clause "New" or "Revised" License

22 stars 8 forks source link

Class limit for datamodules in the multi-label setting #189

Closed reheinrich closed 6 months ago

reheinrich commented 6 months ago

In the multi-label setting, no class limit is used in the datamodules (at least in pretrain_datamodule.py, maybe also in others? --> TODO). This should be fixed so that the class limit is always used if the parameter is provided.