PyTorch循环遍历纪元，然后输出所有纪元的最终值

import torch

# 定义模型和数据集
model = ...
dataset = ...

# 定义损失函数和优化器
criterion = ...
optimizer = ...

# 定义纪元的数量
num_epochs = 10

# 循环遍历纪元
for epoch in range(num_epochs):
    # 初始化纪元的损失值
    epoch_loss = 0.0
    
    # 遍历数据集
    for data in dataset:
        # 前向传播
        inputs, labels = data
        outputs = model(inputs)
        
        # 计算损失
        loss = criterion(outputs, labels)
        
        # 反向传播和优化
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
        
        # 累加纪元的损失值
        epoch_loss += loss.item()
    
    # 输出纪元的最终值
    print("Epoch [{}/{}], Loss: {:.4f}".format(epoch+1, num_epochs, epoch_loss))