@@ -84,7 +84,7 @@ optimizer = AdamW(model.parameters(), lr=2e-5)
loss_fn = torch.nn.CrossEntropyLoss() # 适用于二分类
# 训练循环
-for epoch in range(10):
+for epoch in range(5):
print("开始第{}轮训练".format(epoch + 1))
model.train()
total_loss = 0
