I was trying use the sentiment analysis using trax on some other data. While working, on running training loop I am getting the following error. Can anyone help me with this.
It seems like the data_generator issue. In C3 W1 the data_generator yields a tuple of three - (inputs, targets, example_weights) does your generator do the same?
Thanks a lot @arvyzukai . It seems I really need to work on my debugging skills. Thanks