KevinCDJ
Results
2
comments of
KevinCDJ
当我修改input_point之后遇到了同样的问题，但是我找不到问题出在哪里
我认为我解决了这个问题，当我尝试输出mask_pooling_prototype、cluster_centers、cluster_mask这三个张量运行的设备时，发现cluster_centers运行在cpu上（未修改过代码），然后我在79行报错（我自己是75行）后加入.cuda也不能解决，后发现在计算cluster_centers时，下面加一行 cluster_centers = cluster_centers.cuda()，即可解决