请给一个具体的代码记录模型的训练时间
时间: 2024-12-13 18:27:55 浏览: 10
在Python中,你可以使用`time`模块来简单地记录模型训练的起始和结束时间。下面是一个基本的例子:
```python
import time
# 开始计时
start_time = time.time()
# 这里放置你的模型训练代码
model.fit(x_train, y_train, epochs=epochs)
# 结束计时并计算训练时间
end_time = time.time()
training_duration = end_time - start_time
print(f"模型训练时间:{training_duration}秒")
```
如果你想要更详细的日志记录,可以使用第三方库如`tqdm`配合`datetime`:
```python
from datetime import timedelta
from tqdm.auto import tqdm
start_time = datetime.now()
for epoch in tqdm(range(epochs), desc="Epochs"):
# ...模型训练...
epoch_end_time = datetime.now()
training_duration = epoch_end_time - start_time
print(f"模型训练总耗时:{training_duration}")
```
这会显示每个epoch的进度条,并在结束后显示总训练时间。
阅读全文