fix label models restoring issue from weighted cross entropy

[nithinraok]

Signed-off-by: nithinraok [email protected]

What does this PR do ?

torch creates state dict of weight in weighted cross entropy and while restoring if weight is None, then it throws error. Fixes by introducing weight with all ones.

Collection: ASR

Changelog

• Fixes by introducing weight with all ones.

Before your PR is "Ready for review"

Pre checks:

• [ ] Make sure you read and followed Contributor guidelines
• [ ] Did you write any new necessary tests?
• [ ] Did you add or update any necessary documentation?
• [ ] Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
  • [ ] Reviewer: Does the PR have correct import guards for all optional libraries?

PR Type:

• [ ] New Feature
• [x] Bugfix
• [ ] Documentation