Describe the bug
When I'm using the zarrwriter callback to write extracted features to disc, it breaks since the dictionary returned by the feature encoder is not hasable.
To Reproduce
Load a jit-ed foundational model and run feature extraction.
Expected behavior
This is not clear to me yet. There are multiple ways to handle it.
One way could be to specify what member of the model output dictionary actually needs to be written (cls token or patch embeddings) in the model config.
Other way is to deal with this in the configuration for the zarrwriter.
Environment
dlup version: 0.3.37
How installed: pip install -e .[dev]
Python version: 3.11
Operating System: Linux
Additional context
Screenshots of the model output dictionary and the corresponding error that the writer throws.
Describe the bug When I'm using the zarrwriter callback to write extracted features to disc, it breaks since the dictionary returned by the feature encoder is not hasable.
To Reproduce Load a jit-ed foundational model and run feature extraction.
Expected behavior This is not clear to me yet. There are multiple ways to handle it.
Environment dlup version: 0.3.37 How installed: pip install -e .[dev] Python version: 3.11 Operating System: Linux
Additional context Screenshots of the model output dictionary and the corresponding error that the writer throws.
