Pytorch 中的AverageMeter 造成内存泄漏

解决方法 1

average_meter.update({"key_loss": keynet_loss, "ori_loss": ori_loss, "total_loss": loss  })

改为

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)  })