tensorboard
文章平均质量分 51
活塞君
这个作者很懒,什么都没留下…
展开
-
快速上手-在pytorch使用tensorboardX可视化
安装 不啰嗦 先装tensorflow,然后安装同样版本的tensorboard pip install tensorflow2.2.0 pip install tensorboard2.2.0 导入使用tensorboardX 这里用最简单最常用的summary举例,其他的看官方文档就行了 from tensorboardX import SummaryWriter 然后实例化 writer = SummaryWriter() #不填默认在当前目录下建立runs文件,自己填就是指定存储位置 然后训练过原创 2021-08-06 15:46:05 · 453 阅读 · 0 评论 -
快速上手-pytorch 使用BN,Drop,权重初始化,mini-batch等方式
简介 本文只针对快速上手,BN,Dropout,权重初始化等降低过拟合的方法请自行百度 BN batch norm 在init中初始化BN,1d针对一维、二维输入(加上第一维的batchsize 所以最终输入是2-3维) 2d、3d同理: self.bn = nn.BatchNorm1d(x特征维度) 然后向前传播中加在输入后即可, x = self.bn(x) Dropout 和BN一样 初始化: self.dropout = nn.Dropout(p=0.5) #每次训练0.5的概率丢弃 使用原创 2021-08-06 14:57:29 · 718 阅读 · 0 评论