zilongzhong/SSTN

Program freezing when trying to run on CPU

Jackgihl opened this issue · 1 comments

Hi I am having an issue when trying to run train_ksc.py on a cpu device. I am fairly certain i am commenting out the code running GPU and commenting in the CPU lines but i am still having issues. Has anyone else had success running this program on a cpu device.

Thanks!

I just ended up using Google colab's GPU which worked perfectly. Thank you for making this model public. It is a great resource to study advanced models and increase my knowledge of remote sensing!