Sumit_Vaish:

           self.LABEL.build_vocab(train_ds)


Most likely some of the labels in the test_ds are not recognized. Ideally, the train_ds should contain all the labels expected in test_ds. If not try this:
self.LABEL.build_vocab((train_ds, test_ds))