search

jiguanghe / RISCE

7 stars 4 forks source link

Memory Error #1

Open StanleyLi95 opened 2 years ago

StanleyLi95 commented 2 years ago

Dear Dr. He,

I ran your code of RIS_CE, but my laptop (16-GB RAM, RTX-3070 GPU) reported an error: not enough memory for matrix "A = torch.zeros(N, 2M_rM_RIS, 2N_rN_RIS)". What was the size of your PC RAM when running the code? For 100000 training samples, the code requires 183 GB memory, which needs a 256-GB RAM PC. Did you have a PC with such large RAM?

Thanks Stanley

hejifeng6 commented 2 years ago

我也遇到了同样的问题

StanleyLi95 commented 2 years ago

lishengyuan252@gmail.com, we can discuss this by email if you are interested

hejifeng6 commented 2 years ago

Do you have wechat? If you have wechat we can talk about on it. My wechat ID is hejifeng6. You can add me by my wechat ID.