在复现论文DPGN: Distribution Propagation Graph Network for Few-shot Learning的过程中遇到的问题。
解决方案:代码看起来希望索引在cpu上,那么就在定义它的地方修改device到cpu上就好了
device2 = torch.device('cpu')
class_idx=class_idx.to(device2)
在复现论文DPGN: Distribution Propagation Graph Network for Few-shot Learning的过程中遇到的问题。
解决方案:代码看起来希望索引在cpu上,那么就在定义它的地方修改device到cpu上就好了
device2 = torch.device('cpu')
class_idx=class_idx.to(device2)