def saveModel(new_pt, model, WPQ): """ This functions writes to the protobuf to generate the new .caffemodel """ if DEBUG_Mario: print("\n--- saveModel ---") print("in pt (new_pt): ", new_pt) print("in model: ", model) print("in WPQ: ", WPQ.keys()) net = Net(new_pt, model=model) net.WPQ = WPQ net.finalmodel(save=False) # model new_pt, new_model = net.save(prefix='pruned') if DEBUG_Mario: print("\nout pt (new_pt): ", new_pt) print("out model (new_model): ", new_model) print('\nFinal model ready. For testing you can use:') print('\t$CAFFE_ROOT/build/tools/caffe test -model', new_pt, '-weights', new_model) return new_pt, new_model
def stepend(new_pt, model, WPQ): net = Net(new_pt, model=model) net.WPQ = WPQ net.finalmodel(save=False) # load weights into the caffemodel -by Mario net.dis_memory() #final = net.finalmodel(WPQ, prefix='3r') new_pt, new_model = net.save(prefix='3c') print('caffe test -model',new_pt, '-weights',new_model) return {"final": None}