Thank you for responding!
I’m not sure if this explanation will make sense, but in the array [1, 5, 50], it represents 1 set of 5 words that were converted into their glove vectors. I’m not sure if my batch size would be considered to be 5 or 1. For my label array, [1, 5] represents one set of labels for 5 words.
Could I just change the array to [1, 1, 5, 50]?
I am a bit confused as to how I can approach this, and any further clarification would be extremely helpful!