Issue with Introspective Rationale Explainer for BERT on text

[ahana17]

explainer.fit(df_train, df_test) is giving error
'RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.'
I'm getting this error on running the example notebook as well as trying on my own data