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}