如何使用matplotlib显示pytorch的损失图和准确度图

pip install matplotlib
pip install torch

import matplotlib.pyplot as plt
import torch

losses = []
accuracies = []

for epoch in range(num_epochs):
    # 训练代码
    
    # 在每个epoch结束时计算并记录损失和准确度
    loss = calculate_loss(...)
    accuracy = calculate_accuracy(...)
    losses.append(loss)
    accuracies.append(accuracy)

plt.figure(figsize=(10, 5))
plt.plot(losses, label='Loss')
plt.xlabel('Epoch')
plt.ylabel('Loss')
plt.title('Training Loss')
plt.legend()
plt.show()

plt.figure(figsize=(10, 5))
plt.plot(accuracies, label='Accuracy')
plt.xlabel('Epoch')
plt.ylabel('Accuracy')
plt.title('Training Accuracy')
plt.legend()
plt.show()