pytorch重载ddp模型报错RuntimeError: storage has wrong size: expected -4825104973866335134 got 2048
报错内容RuntimeError: storage has wrong size: expected -4825104973866335134 got 2048
解决方案在保存模型之前，增加 rank==0的判定，
if rank == 0：
torch.save(model.module.state_dict(), "./weights/model-best.pth")
