在上面的代码中加入打印模型在训练集的准确率的相关代码

可以在训练过程中加入以下代码来打印模型在训练集上的准确率：

train_stats = evaluate(data_loader_train, model, device)
print(f"Accuracy of the network on the {len(dataset_train)} training images: {train_stats['acc1']:.1f}%")

其中，`data_loader_train`是训练集数据的 DataLoader，`dataset_train`是训练集数据集，`evaluate`函数同样可以用来在训练集上评估模型性能。