Removed the hardcoded classes length value of 10. Replaced it with args.num_classes. Now this allows us to experiment with this model across different datasets, without errors.
The hardcoded value, throws an error when used with other class length with the embedding layer.
Removed the hardcoded classes length value of 10. Replaced it with args.num_classes. Now this allows us to experiment with this model across different datasets, without errors.
The hardcoded value, throws an error when used with other class length with the embedding layer.