search

rctn / sparsecoding

Reference sparse coding implementations for efficient learning and inference.
BSD 3-Clause "New" or "Revised" License
17 stars 4 forks source link

bug fix where LMS failed on cuda device. ones-array ref. grads to cas… #10

Closed belsten closed 2 years ago

belsten commented 2 years ago

…t to device. changed instantiation of coefficients to correct device instead of case. Avoids error with non-leaf autograd stuff

Also minor syntax changes/whitespace

belsten commented 2 years ago

Fixed!

denkle commented 2 years ago

Awesome!