文章浏览阅读4.7k次，点赞3次，收藏8次。跟踪并保存训练过程中的损失函数以CornerNet为例，他的源码并没有使用损失函数可视化的功能，有时候需要查看损失函数的变化趋势来确定超参。那么此时就需要手动去记录并储存损失函数值在train.py中，关于损失函数的部分如下所示：with stdout_to_tqdm() as save_stdout:for iteration in tqdm(range(start_iter + 1, max_..._训练次数和损失函数曲线