在Pytorch中使用tensorboard

最新推荐文章于 2024-08-19 15:45:05 发布

Huntersts

最新推荐文章于 2024-08-19 15:45:05 发布

阅读量522

点赞数

本文链接：https://blog.csdn.net/qq_35260390/article/details/105157800

版权

本文简要介绍如何在Pytorch中利用Tensorboard来查看网络结构和训练损失，过程与TensorFlow类似。

摘要由CSDN通过智能技术生成

参照官方文档的一部分

在pytorch中使用tensorboard的方法也挺简单，如果要看网络结构和训练损失的话基本上和在TF中使用是差不多的。

import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import torch.utils.data as data

# 前提要在自己的环境中安装tensorboard 直接pip安装就行
from torch.utils.tensorboard import SummaryWriter
class Net(nn.Module):
	def __init__(self):
		super(Net,self).__init__()
		self.conv1 = nn.Conv2d(1, 6, 5)
        self.pool1 = nn.MaxPool2d(2, 2)
        self.conv2