search

janclemenslab / das

Deep Audio Segmenter
http://janclemenslab.org/das/
Apache License 2.0
28 stars 10 forks source link

das.train.train error - unexpected keyword argument "sample_weight_mode" #80

Closed emory-saba closed 1 week ago

emory-saba commented 1 week ago

Hello,

I am attempting to train a DAS model in Google Colab and repeatedly receive the error: "Trainer.compile() got an unexpected keyword argument 'sample_weight_mode'".

Screen Shot 2024-11-18 at 4 32 45 PM

The Google Colab AI assistant suggests that this error is likely arising from the fact that das.models.tcn_stft calls modile.compile with sample_weight_mode, which was used in older versions of Keras but has been removed in newer versions.

postpop commented 1 week ago

Hi, thanks for raising this issue. I agree - it's probably an issue with the Keras version. Can you post the output of das version?

emory-saba commented 1 week ago

Thanks for the quick response! The version I have on my Colab is 0.32.4

postpop commented 1 week ago

I've fixed the version of tensorflow that is installed with DAS.

The install command in the notebook is now !pip install das "tensorflow<2.16"

Tthe updated notebook is now online.

Please reopen the issue if the problem persists.