I fixed it. It needed some code changes. The problem was in handling sequence length. The updated code is in the linked notebook.
3 Likes
I fixed it. It needed some code changes. The problem was in handling sequence length. The updated code is in the linked notebook.