search

keunwoochoi / music-auto_tagging-keras

Music auto-tagging models and trained weights in keras/theano
MIT License
614 stars 142 forks source link

Loss function and Optimizer #8

Closed jayavanth closed 7 years ago

jayavanth commented 7 years ago

What kind of loss function and what kind of optimizer are you using for this? I couldn't find them in the code. According to the paper, it is cross entropy and adam resp.

keunwoochoi commented 7 years ago

That's correct, cross entropy and adam.

jayavanth commented 7 years ago

I didn't see any lines in the code referring to the loss function or optimizer. May I know where in the code you're using those? Thanks.

keunwoochoi commented 7 years ago

Hm, there's no lines for those in this repo. Adding them in Keras is trivial.