pytorch训练模型计时

在PyTorch中，可以使用time模块来计时训练模型的时间。下面是一个示例代码：

import time

start = time.time()  # 开始计时

# 训练模型的代码

end = time.time()  # 结束计时

duration = end - start  # 计算训练时间

print("训练模型的时间为：", duration, "秒")

在这个示例中，我们使用了time模块的time函数来获取当前时间。在开始训练模型之前调用start = time.time()，在训练结束后调用end = time.time()，然后计算时间差得到训练时间。最后，我们将训练时间打印出来。

请注意，这只是一个示例代码，你需要根据你的具体情况来修改和适应。同时，你还可以根据需要在计时开始和结束之间添加其他代码，比如加载数据、定义模型、定义损失函数、定义优化器等。

引用[3]中的代码片段给出了一个更完整的示例，其中使用了time模块来计时，并且在计时开始和结束之间进行了一些其他操作。你可以根据需要参考这个示例来计时你的训练模型过程。