google-deepmind/dm-haiku

Orthogonal initilizer does not support MLP?

rezunli96 opened this issue · 1 comments

Hi, one question about the orthogonal initilizer, so it turns out if we initialize an MLP net with it, it will have the following error:

https://github.com/deepmind/dm-haiku/blob/edfdcc3457c781759c87a0a08921c5b3f85d9a45/haiku/_src/initializers.py#L277-L279

But this contradicts to my understanding: since a linear layer can be represented as some kind of matrix, why orthogonal initilizer cannot work in this case?

Sorry, turns out to be a bug on my side. Thanks.