解决Windows中caffe用python运行solve时kernel restart问题

捣鼓了两天,本来以为只要把python和caffe配置好,那么在运行时候应该可以在任何工作目录下了。

但是,在Windows中,DOS默认的工作路径下运行是不行的,要加载到caffe的根目录下面才行,和用bat,Ubuntu下情况类似。

这样再运行应该就没问题了:

import caffe
caffe.set_mode_cpu()
solve=caffe.SGDSolver('D:\\caffe-master\\examples\\fruit\\solver.prototxt')

运行试验,在doc中显示的就和在Ubuntu和bat处理的一样了
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章