解决方法 1
average_meter.update({"key_loss": keynet_loss, "ori_loss": ori_loss, "total_loss": loss })
改为文章来源地址https://www.toymoban.com/news/detail-707657.html
average_meter.update({"key_loss": keynet_loss.detach().item(), "ori_loss": ori_loss.detach().item(), "total_loss": loss.detach().item() })
解决方法 2
average_meter.update({"key_loss": torch.tensor(keynet_loss), "ori_loss": torch.tensor(ori_loss), "total_loss": torch.tensor(loss) })
文章来源：https://www.toymoban.com/news/detail-707657.html
到了这里，关于Pytorch 中的AverageMeter 造成内存泄漏的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
