如何查看模型的参数
Created by: sguuaa
在fluid.io.load_vars后,希望看看模型的各个参数值,不知道如何debug。
##============= net =============##
with fluid.program_guard(tp, sp):
img = fluid.layers.data(
name='img', shape=[3, input_size[1], input_size[0]], dtype='float32')
label = fluid.layers.data(name='label', shape=[1], dtype='int32')
predict = net(img, args.num_classes)
##=========== exe ===================##
place = fluid.CPUPlace()
if args.gpu != '-1':
place = fluid.CUDAPlace(0)
exe = fluid.Executor(place)
exe.run(sp)
##========== load model ===============##
if args.init_weights_path:
logging.info("load from:" + args.init_weights_path)
assert (os.path.isdir(args.init_weights_path))
def if_exist(var):
return os.path.exists(os.path.join(args.init_weights_path, var.name))
fluid.io.load_vars(exe, dirname=args.init_weights_path, main_program=tp, predicate=if_exist)