Program freezing when trying to run on CPU

zilongzhong / SSTN

PyTorch Implementation of SSTNs for hyperspectral image classifications from the IEEE T-GRS paper "Spectral-Spatial Transformer Network for Hyperspectral Image Classification: A FAS Framework."

106 stars 23 forks source link

Program freezing when trying to run on CPU #6

Closed Jackgihl closed 2 years ago

Jackgihl commented 2 years ago

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!

Jackgihl commented 2 years ago

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!