目录
1. 目的
利用tensorboard实时的跟进训练情况,可视化当前的预测结果,和参与的数据增强后的训练数据情况。
2. 代码
if self.every_n_inner_iters(runner, self._print_interval_iter):
# loss
loss = runner.batch_iter_output['batch_loss']
prob_dice_loss = runner.batch_iter_output['prob_dice_loss']
thres_dice_loss = runner.batch_iter_output['thres_dice_loss&#