pytorch
walking_visitor
想起什么写什么
展开
-
pytorch用法笔记--不定期更新
torch.matmul(a,a.T) 矩阵乘以它的转置矩阵,如3*4,4*3,相乘后结果是一个3*3的对称矩阵 torch.diag(a, 2) 取以对角线为起点的元素,对角线的位置为0,2代表从对角线开始往上的第2条对角线,也可以是负值,也就是对角线往下 torch.cat((a, b), dim=0) 拼接两个矩阵,按照dim=0的维度,比如3*4,3*5,拼接后就是3*9,dim=0也就是横向拼接,dim=1也就是纵向拼接 np.fill_diagonal(a,0)/torch.fill原创 2021-09-24 09:47:57 · 331 阅读 · 0 评论 -
pytorch问题笔记--不定期更新
loss为nan 搭建了基于restnet50的多损失行人重识别程序,通过restnet50提取图片特征,在迭代训练过程中出现了LOSS为NAN的情况,观察日志发现最初的几个bitch_size所有的一切都是正常的,从restnet50提取的特征也都是正常的,但在几轮之后,从restnet50提取的特征变开始变成nan。 造成nan的问题有几种可能: 图片样本的问题--在本次训练中全都同样规格的样本,故排除这个问题 程序的问题,如某些地方的除法或者log运算导致了nan---认真检查程序没有此问题原创 2021-09-24 09:39:10 · 118 阅读 · 0 评论