可视化日志

参考:

[PyTorch]Tensorboard可视化实现

[PyTorch]Tensorboard使用实践

使用TensorboardX进行训练过程可视化,记录了训练过程中损失值学习率的变化,并且记录每个阶段的测试结果(AP)

新建了一个MetricLogger类,用于平滑损失值,预防单个极大损失值影响损失值可视化

相关文件

  • py/ssd/engine/trainer.py
  • py/ssd/utils/metric_logger.py