a error when I run sample.py

[beautyMJ]

When I run the sample.py, an error occurred: Traceback (most recent call last): File "sample.py", line 57, in model = Model(saved_args, True) File "C:\YMJ\download_pro\magenta-master\magenta\models\sketch_rnn\sketch-rnn-master\model.py", line 93, in init outputs, states = custom_rnn_autodecoder(inputs, self.initial_input, self.initial_state, cell, scope='rnn_mdn') File "C:\YMJ\download_pro\magenta-master\magenta\models\sketch_rnn\sketch-rnn-master\model.py", line 71, in custom_rnn_autodecoder output, new_state = cell(inp, states[-1]) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 183, in call return super(RNNCell, self).call(inputs, state) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\layers\base.py", line 575, in call outputs = self.call(inputs, *args, **kwargs) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 1066, in call cur_inp, new_state = cell(cur_inp, cur_state) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 183, in call return super(RNNCell, self).call(inputs, state) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\layers\base.py", line 575, in call outputs = self.call(inputs, *args, **kwargs) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 441, in call value=self._linear([inputs, h]), num_or_size_splits=4, axis=1) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 1189, in call res = math_ops.matmul(array_ops.concat(args, 1), self._weights) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\math_ops.py", line 1891, in matmul a, b, transpose_a=transpose_a, transpose_b=transpose_b, name=name) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\gen_math_ops.py", line 2436, in _mat_mul name=name) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\op_def_library.py", line 787, in _apply_op_helper op_def=op_def) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\ops.py", line 2958, in create_op set_shapes_for_outputs(ret) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\ops.py", line 2209, in set_shapes_for_outputs shapes = shape_func(op) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\ops.py", line 2159, in call_with_requiring return call_cpp_shape_fn(op, require_shape_fn=True) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\common_shapes.py", line 627, in call_cpp_shape_fn require_shape_fn) File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\common_shapes.py", line 691, in _call_cpp_shape_fn_impl raise ValueError(err.message) ValueError: Dimensions must be equal, but are 512 and 261 for 'rnn_mdn_1/rnn_mdn/multi_rnn_cell/cell_0/cell_0/basic_lstm_cell/MatMul_1' (op: 'MatMul') with input shapes: [1,512], [261,1024]. Could you tell me how to solve it? Thank you very much!