search

CalculatedContent / WeightWatcher

The WeightWatcher tool for predicting the accuracy of Deep Neural Networks
Apache License 2.0
1.47k stars 125 forks source link

Add support for TF Multihead attention and TimeDistributed layers #70

Open charlesmartin14 opened 3 years ago

charlesmartin14 commented 3 years ago

Add support for TF Multihead attention and TimeDistributed layers This seems to be broken in the latest version..one we moved to strong typing for the layer types, we now have to support complex typed layers like this explicitly

charlesmartin14 commented 3 years ago

I think I fixed this...should be ready in the 0.49 release