Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False
此时改为:
torch.load("0.9472_0048.weights",map_location='cpu')
Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False
此时改为:
torch.load("0.9472_0048.weights",map_location='cpu')