gpauloski/kfac-pytorch

Add diagonal approximation for linear layers

Opened this issue · 0 comments

Add diagonal approximation for linear layers