loss function for the prado model

Prerequisites

Please answer the following questions for yourself before submitting an issue.

• [yes ] I am using the latest TensorFlow Model Garden release and TensorFlow 2.
• [yes] I am reporting the issue to the correct repository. (Model Garden official or research directory)
• [yes] I checked to make sure that this issue has not been filed already.

1. The entire URL of the file you are using

https://github.com/tensorflow/models/tree/master/research/sequence_projection

2. Describe the bug

Rank mismatch: Rank of labels (received 2) should equal rank of logits minus 1 (received2)

3. Steps to reproduce

I tried to run prado model on my own single label dataset. I created a text file and change the multilabel option into single, but after I run this, it shows the rank mismatch. I checked the code, it seems like the loss function requires a mismatched shape for input values.

4. Expected behavior

It expected to train the data.

6. System information

• OS Platform and Distribution: Linux Ubuntu 16.04
• TensorFlow installed from (source or binary): 2.3
• TensorFlow version (use command below): 2.3
• Python version: 3.6
• Bazel version (if compiling from source): 3.5

Also, I tried to change the multilabel into true, and it works. But when I tried to evaluated my data, it takes long long time and without any outcomes. I am not sure what happened in here.