Nan loss and accuracy for capsnet implementation for custom dataset

ageron / handson-ml

⛔️ DEPRECATED – See https://github.com/ageron/handson-ml3 instead.

Apache License 2.0

25.18k stars 12.92k forks source link

Nan loss and accuracy for capsnet implementation for custom dataset #538

Open codehard123 opened 4 years ago

codehard123 commented 4 years ago

I tried your capsule networks implementation on a custom(GTZAN) data set by classifying images of spectrograms. But I Get loss as Nan . I could not resolve this issue. Any ideas for nan loss?