TensorBoard 使用教程 3taming-3dgs中TensorBoard的使用代码日志文件查看TensorBoard代码与TensorBoard对应关系taming-3dgs中TensorBoard的使用代码fromtorch.utils.tensorboardimportSummaryWriter tb_writer=SummaryWriter(args.model_path)deftraining_report(tb_writer,iteration,Ll1,loss,l1_loss,elapsed,testing_iterations,scene:Scene,renderFunc,renderArgs):""" 训练报告函数，记录训练日志和评估指标 参数: tb_writer: Tensorboard写入器 iteration: 当前迭代次数 Ll1: L1损失值 loss: 总损失值 l1_loss: L1损失函数 elapsed: 当前迭代耗时 testing_iterations: 测试迭代次数列表 scene: 场景对象 renderFunc: 渲染函数 renderArgs: 渲染参数 """# 记录训练损失到Tensorboardiftb_writer:tb_writer.add_scalar('train_loss_patches/l1_loss',Ll1.item(),iteration)tb_writer.add_scalar('train_loss_patches/total_loss',loss.item(),iteration)tb_writer.add_scalar('iter_time',elapsed,iteration)# 在指定迭代进行测试评估ifiterationintesting_iterations:torch.cuda.empty_cache()validation_configs=({'name':'test','cameras':scene.getTestCameras()},{'name':'train','cameras':[scene.getTrainCameras()[idx%len(scene.getTrainCameras())]foridxinrange(5,30,5)]})forconfigin