- 博客(2)
- 收藏
- 关注
RSS订阅原创 tensorflow测试时读取任意指定的check point的网络参数
tensorflow在训练时会保存三个文件，model.ckpt-xxx.data-00000-of-00001model.ckpt-xxx.indexmodel.ckpt-xxx.meta其中第一个储存网络参数值，第二个储存每一层的名字，第三个储存图结构随着训练的过程，每隔一段时间都会保存一组以上三个文件，而在训练之前我们并不知道什么时候可以达到最佳的拟合，训练时间过短会导致欠...
2018-08-23 10:47:46
2457
原创 Pytorch RuntimeError: should contain 1 elements not 64/ValueError: expected 4D input (got 2D input)
在运行pytorh的过程中一次产生了如题两个错误，错误来自于nn.BatchNorm2d(N)，其中N=channels然而输入必须是一个四维Tensor，在我的程序中输入Tensor是一个1*47的向量，一个batchsize=100，所以起初的输入是一个100*47的Tensor，然而pytorch是不识别这种输入的，解决办法：将输入重新排列input = input.view(...
2018-08-23 03:45:07
6881
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人