-
Notifications
You must be signed in to change notification settings - Fork 22
About the Dictionary Learning #2
Description
Hi, minhnhat93. I tested your implementation of the LISTA. During the dictionary learning part, I find some bugs. The dictionary couldn't be learned when I tried to fix them. Could you please help me with it?
Besides, according to the original paper of LISTA in parameter setting, the learning rate for LISTA training decays as 1/n (n-iter number), but the thing is when I did the exact same thing. The gradient is exploded to NaN. I guess it is the single-sample training strategy instead of mini-batch training causes this, making the gradient unstable. I notice you set extremely small learning rate, so there is no such problem. My question is if it is possible to deploy batch gradient descends to address this problem?
Looking forword to hearing from you!