Error ValueError: too many values to unpack (expected 2) in model training

[John6666]

Maybe this?

Batch_size, seq_length = input_shape ValueError: too many values to unpack (expected 2) Transformer Sentence Similarity Classification

I encountering a problem when doing Sentence Similarity Classification Task Using Transformer. It happens when I was about to train my model, I load and process my dataset in this way, and this is what the training dataset element shape looks like: [image] but when I train the model, it says: ValueError: Caught ValueError in replica 0 on device 0. Original Traceback (most recent call last): File “/home/fanshaoqi/anaconda3/envs/tf2/lib/python3.10/site-packages/torch/nn/parallel/parallel_ap…