Giters

nshaud / DeepHyperX

Deep learning toolbox based on PyTorch for hyperspectral data classification.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

Heboxian opened this issue · comments

commented

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