How could fix it ? Something wrong when I run this code.

[ysgncss]

Traceback (most recent call last):
File "codes/train.py", line 37, in
text_capsnet.train()
File "C:\Users\86156\Desktop\text-capsule-network-master\codes\models\text_capsnet.py", line 62, in train
callbacks=[lr_scheduler, ckpt_callback])
File "D:\software\python36\Lib\site-packages\keras\engine\training.py", line 1239, in fit
validation_freq=validation_freq)
File "D:\software\python36\Lib\site-packages\keras\engine\training_arrays.py", line 196, in fit_loop
outs = fit_function(ins_batch)
File "D:\software\python36\Lib\site-packages\tensorflow_core\python\keras\backend.py", line 3476, in call
run_metadata=self.run_metadata)
File "D:\software\python36\Lib\site-packages\tensorflow_core\python\client\session.py", line 1472, in call
run_metadata_ptr)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Incompatible shapes: [40,2,6] vs. [40,2]
[[{{node loss/capsule_norm_1_loss/margin_loss/mul_2}}]]