![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch框架学习
hongxing.Li
这个作者很懒,什么都没留下…
展开
-
Pytorch---常用函数
函数列表torch.flatten()函数nn.nn.Linear()函数 torch.flatten()函数 A = torch.tensor([[[1,2,3,4],[5,6,7,8],[9,10,11,12]],[[13,14,15,16],[17,18,19,20],[21,22,23,24]]]) out1 = torch.flatten(A) out2 = torch.flatten(A,0) out3 = torch.flatten(A,1) print("默认 = ",out1) print原创 2022-05-23 16:46:49 · 180 阅读 · 0 评论 -
Pytorch----DarkNet53网络
文章目录1. DarkNet网络结构1.1 卷积单元----Convolutional1.2 Residual二、Pytorch实现三、Tensorboard绘制网络结构图 1. DarkNet网络结构 Yolov3、Yolov4的backbone采用了CSP_DarkNet_53网络,其激活函数采用Mish。本文章复现DarkNet_53激活函数采用LeakyReLu,两者结构基本一致。 1.1 卷积单元----Convolutional Convolutional卷积单元由Conv+BN+LeakyRe原创 2022-05-23 11:19:21 · 617 阅读 · 0 评论 -
pytorch-模型保存
pytorch框架下两种模型保存及加载方式 方法1 以字典(key-value)保存模型的参数 torch.save(vgg16.state_dict(), "路径") vgg16.load_state_dict(torch.load("路径")) 方法2 保存模型参数及模型结构 torch.save(vgg16, "路径h") model = torch.load("路径") ...原创 2022-03-15 23:22:27 · 169 阅读 · 0 评论