yolox训练得到的train_log.txt,怎么用tensorboard查看total_loss曲线,需要怎么修改trainer.py文件
时间: 2023-06-16 14:07:55 浏览: 234
使用Tensorboard工具查看Loss损失率
5星 · 资源好评率100%
如果你使用的是YOLOX的官方代码,那么在训练过程中,YOLOX会自动保存训练日志到`exp/exp_name/train_log.txt`文件中。要使用TensorBoard可视化训练日志,可以按照以下步骤进行操作:
1. 在YOLOX根目录下打开终端,输入以下命令启动TensorBoard:
```
tensorboard --logdir exp/exp_name
```
其中,`exp/exp_name`是你的YOLOX实验结果的保存路径。
2. 打开浏览器,在地址栏输入`http://localhost:6006`,进入TensorBoard网页。
3. 在TensorBoard网页左侧的菜单栏中,找到`Scalars`选项卡,点击进入。
4. 在`Scalars`页面中,将`total_loss`输入到搜索框中,即可查看`total_loss`的曲线。
如果你希望修改YOLOX代码以便在训练时自动保存TensorBoard日志,可以按照以下步骤进行操作:
1. 在YOLOX根目录下找到`yolox/core/trainer.py`文件,打开该文件。
2. 在`Trainer`类的`__init__`方法中,添加以下代码:
```
self.writer = SummaryWriter(log_dir=self.exp_log_dir)
```
这将会创建一个`SummaryWriter`对象,用于保存TensorBoard日志。
3. 在`Trainer`类的`train`方法中,找到以下代码:
```
loss, log_vars = self.forward_backward(batch)
```
4. 在该代码后面添加以下代码:
```
self.writer.add_scalar('train/total_loss', loss.item(), self.iter)
```
这将会在每个训练步骤结束后将`total_loss`的值写入TensorBoard日志。
5. 最后,在训练结束后,记得关闭`SummaryWriter`对象,以便将日志保存到磁盘:
```
self.writer.close()
```
这些修改将会让YOLOX在训练时自动保存TensorBoard日志,方便你查看训练过程中的各项指标。
阅读全文