neurostatslab / vocalocator

Deep neural networks for sound source localization and vocalization attribution.
MIT License
2 stars 0 forks source link

`run_eval` function in `main.py` brittle to output shape #21

Closed amanchoudhri closed 1 year ago

amanchoudhri commented 1 year ago

On line 147 of main.py inside the run_eval function, the code assumes that the output shape from the model will be (3, 2), but this will fail for point predictors, ensembles, or other more complex models.