Hi Loukas,
When I did audio classification, I tried to leave all preprocessing (like librosa and correlated) on preprocessing layers, maybe restructuring the pipeline might fix: Transfer learning with YAMNet for environmental sound classification 5