An error is always reported when the victim model is xlnet_sst and the data set sample is sst

[wei1826676931]

I want to try to use the existing xlnet_sst as the attacked model. Unfortunately, it keep reporting errors. Can you check and try this example？thanks!

[a710128]

I reproduced the error. This is because XLNET uses the second dimension to represent batch, while most other models use the first dimension.

XLNET: (seq_length, batch_size, hidden_size)
OTHER: (batch_size, seq_length, hidden_size)

We will fix this issue in the next release.

[wei1826676931]

辛苦了，谢谢~