XiaoHeiBlack的专栏

原创 pytorch 若干小坑

caffe灵活性还是比不上torch，想要快速建立新模型，torch是个不错的选择 #batch size 设置略大段错误自己练手写的lenet和官方的lennet[https://github.com/pytorch/examples/blob/master/mnist/main.py]在CPU中运行时，设置稍微大一点的batch size(8以上)，就会出现段错误，而且也没错误栈