问题描述
在训练Pytorch模型的时候,报错
Traceback (most recent call last):
File "/home/visionx/nickle/temp/SimCLR/linear_evaluation.py", line 304, in <module>
loss_epoch, accuracy_epoch = train(
^^^^^^
File "/home/visionx/nickle/temp/SimCLR/linear_evaluation.py", line 124, in train
plt.scatter(t_sne_embeddings[ node_labels == class_id, 0],
~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/visionx/anaconda3/envs/simclr-pt/lib/python3.11/site-packages/torch/_tensor.py", line 970, in __array__
return self.numpy()
^^^^^^^^^^^^
TypeError: can't convert cuda:0 devic


997

被折叠的 条评论
为什么被折叠?



