解决方法 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) })