Mr_WHITE2的博客

pytorch加载模型时报错RuntimeError: Error(s) in loading state_dict for DataParallel:Missing key(s) in state_dict: “module.backbone.layers.0.stage_1.layers.0.weight”,这是因为加载的预训练模型之前使用了torch.nn.DataParallel()，而此时没有使用，所以可以加上该模块或者去掉。解决方法一：加上torch.nn.DataParallel()模

1.numpy转为tensornp2tensor = torch.fromnumpy(numpy1) # numpy1为ndarray2.tensor转为numpy tensor2np = tensor1.numpy() # tensor1为tensor但是有可能会有这个错误：TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor那么将tensor先拷贝到CPU再转 tenso