checkpoint loading and logit masking

DBD-research-group / BirdSet

A benchmark dataset collection for bird sound classification

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

22 stars 8 forks source link

Closed Moritz-Wirth closed 7 months ago

Moritz-Wirth commented 7 months ago

load checkpoints that were saved from callback: new hydra parameter: module.network.model.local_checkpoint

masking logits from pretrained models: module.network.model.pretrain_info.hf_pretrain_name

eg.: XCM loads the whole model with len(xcm_classes) but masks to classes from testing dataset