Traceback (most recent call last):


 "/Users/necula/Source/jax/jax/interpreters/xla.py", line 230, in primitive_computation

return c.Build()


 "/Users/necula/Source/jax/jax/lib/xla_bridge.py", line 281, in Build

*args, **kwargs)


 "/Users/necula/Source/jax/build/jaxlib/xla_client.py", line 734, in Build

return Computation(self._builder.Build(), backend=backend)
RuntimeError: Invalid argument: Slice size at index 0 in gather op is out of range, must be within [0, 6), got 10.: 

Traceback (most recent call last):


 "train.py", line 135, in <module>

test_abs(args, device_id, cp, step)


 "E:\project\PreSumm\src\train_abstractive.py", line 215, in test_abs

model = AbsSummarizer(args, device, checkpoint)


 "E:\project\PreSumm\src\models\model_builder.py", line 217, in __init__

self.load_state_dict(checkpoint['model'], strict=True)


 "C:\Users\Admin\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 769, in load_state_dict

self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for AbsSummarizer:
