Inline documentation for trainer_state_model.py

[madhhuurrii]

Description

Inline documentation for trainer_state_model.py

Relevant Issue

Issue #189

Affected Dependencies

NA

Checklist

• [x] I have followed the Contribution Guidelines and Code of Conduct
• [x] I have commented my code following the NimbleEdge Styleguide
• [ ] I have labeled this PR with the relevant Type labels
• [ ] My changes are covered by tests

[netlify[bot]]

✅ Deploy Preview for goofy-agnesi-4565cb canceled.

Name	Link
🔨 Latest commit	0d398f7d83e07716c0f3dda4f12cd6986a19bb50
🔍 Latest deploy log	https://app.netlify.com/sites/goofy-agnesi-4565cb/deploys/62740107983dad00086f9913