- 博客(1)
- 收藏
- 关注
原创 pytorch如何查看网络结构所有的权值和偏置
pytorch如何查看网络结构所有的权值和偏置由于需要将神经网络的参数导出,在硬件上实现,网络结构的参数总是显示一部分,省略了大部分,查阅了很多文章,总是没有很好的解决办法,无意中发现了一条命令,试了一下,果然很有效,能够打印出所有的参数。torch.set_printoptions(profile=“full”)默认是省略的abc=open(‘parameter.txt’,‘w’)para=list(net.parameters())print(para)print(para,file=ab
2020-08-17 09:54:41
2210
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人