Closed jenniew closed 6 years ago
I think caffe model save only support graph model. @wzhongyuan
@yiheng yes, but seems this loaded model is a Graph, I will take a look
@wzhongyuan I see a sequential container in the error message.
@jenniew
I did try and ran below code successfully
val model = CaffeLoader.loadCaffe[Float]("/home/jerry/lab/data/caffe/googlenet/deploy.prototxt",
"/home/jerry/lab/data/caffe/googlenet/bvlc_googlenet.caffemodel", null)
._1.asInstanceOf[Graph[Float]]
CaffePersister.persist[Float]("/tmp/gle.prototxt", "/tmp/gle.caffemodel", model, true, true)
did you modify the model ?
val newModel = Inception_v1_NoAuxClassifier(param.classNumber) val model = Module.loadCaffe(newModel, param.prototxt, param.modelSnapshot, matchAll = false) After trained, cannot save to caffe model or bigdl model. I used pretrained place-365 googlenet model from https://github.com/CSAILVision/places365
can we close the issue now? @wzhongyuan @jenniew
@wzhongyuan any update of this?
@yiheng I did try with above code without any issue, I think it's sequential issue, we can close it for now
Load a caffe googlenet model to bigdl model, fine tune the model, and save back to caffe mode, and get error. Code: model.saveCaffe("/model/front_story_40/a.prototxt", "/model/front_story_40/a.caffemodel")
Log: test_story.txt