search

nshaud / DeepHyperX

Deep learning toolbox based on PyTorch for hyperspectral data classification.
Other
439 stars 122 forks source link

expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight #54

Open Heboxian opened 2 years ago

Heboxian commented 2 years ago

when I run "python main.py --model hamida --dataset PaviaU --training_sample 0.5 --patch_size 7 --epoch 10 --cuda 1 " it has RuntimeError: :expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight