目录
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']
thres_l1_loss = runner.batch_iter_output['thres_l1_loss']
summary_fun.add_scalar('batch_loss', loss, iters)
summary_fun.add_scalar('prob_dice_loss', prob_dice_loss, iters)
summary_fun.add_scalar('thres_dice_loss', thres_dice_loss, iters)
summary_fun.add_scalar('thres_l1_loss', thres_l1_loss, iters)
# orig image and label
target = runner.data_batch[1] # list
input_tensor = runner.data_batch[0] # torch.Size([8, 3, 512,