agent保存之后加载报尺寸不匹配错误
Created by: liucc09
代码如下:
agent.save(ckpt)
agent.restore(ckpt)
报错:
---------------------------------------------------------------------------RuntimeError Traceback (most recent call last) in 1 print(ckpt) 2 agent.save(ckpt) ----> 3 agent.restore(ckpt) /opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/parl/core/fluid/agent.py in restore(self, save_path, program) 194 dirname=dirname, 195 main_program=program, --> 196 filename=filename) /opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/io.py in load_params(executor, dirname, main_program, filename) 796 main_program=main_program, 797 predicate=is_parameter, --> 798 filename=filename) 799 800 /opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/io.py in load_vars(executor, dirname, main_program, vars, predicate, filename) 680 main_program=main_program, 681 vars=list(filter(predicate, main_program.list_vars())), --> 682 filename=filename) 683 else: 684 load_prog = Program() /opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/io.py in load_vars(executor, dirname, main_program, vars, predicate, filename) 739 "Shape not matching: the Program requires a parameter with a shape of ({}), " 740 "while the loaded parameter (namely [ {} ]) has a shape of ({}).". --> 741 format(orig_shape, each_var.name, new_shape)) 742 743 RuntimeError: Shape not matching: the Program requires a parameter with a shape of ((16, 64)), while the loaded parameter (namely [ fc.w_18 ]) has a shape of ((64, 64)).