怎么把训练过程的每个epoch中print的内容保存

您可以使用Python的logging模块来记录训练过程中的print输出，将输出写入日志文件中。

以下是一个简单的例子：

import logging

# 创建logger对象
logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)

# 创建文件handler，将日志写入文件中
file_handler = logging.FileHandler('training.log')
file_handler.setLevel(logging.INFO)

# 创建控制台handler，将日志输出到控制台
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.INFO)

# 定义日志格式
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
file_handler.setFormatter(formatter)
console_handler.setFormatter(formatter)

# 将handler添加到logger对象中
logger.addHandler(file_handler)
logger.addHandler(console_handler)

# 在训练过程中使用logger输出信息
for epoch in range(num_epochs):
    loss = train_one_epoch(model, optimizer, train_loader, device)
    logger.info('Epoch: {}, Loss: {}'.format(epoch, loss))

这个例子中，我们创建了一个名为“training”的logger对象，并将其输出级别设置为INFO。我们创建了两个handler，一个是文件handler，将日志写入文件“training.log”中，另一个是控制台handler，将日志输出到控制台。我们为handler设置了输出格式，并将它们添加到logger对象中。最后，在训练过程中，我们使用logger对象输出信息，这些信息将会被记录到日志文件中。