search

nmi-lab / decolle-public

GNU General Public License v3.0
41 stars 22 forks source link

CUDA error on tutorial #3

Open tasbolat1 opened 4 years ago

tasbolat1 commented 4 years ago

Hello everyone, while running tutorial on classification using dcll (second tutorial), I have encountered the following error during training very simple two layer mlp:

~/some_python_examples/VT_SNN/auxillary_files/../../pytorch-lif-autograd/decolle_public/decolle/base_model.py in forward(self, Sin_t)
    210         #print('Sin_t:', Sin_t)
    211 
--> 212         Q = self.beta * state.Q + self.tau_s * Sin_t
    213         P = self.alpha * state.P + self.tau_m * state.Q  # TODO check with Emre: Q or state.Q?
    214         R = self.alpharp * state.R - state.S * self.wrp

RuntimeError: CUDA error: an illegal memory access was encountered

I googled possible solutions, but with no success. Is there anything I am missing while setting DECOLLE?

Thanks, Tas

eneftci commented 4 years ago

I remember having these issues when having problems. I updated this version with the latest version. Please check if you have the same error now. If you're following the tutorial on pytorch-lif-autograd, note that there was a typo in the MLP parameters which prevented learning.