added doc for trainer.py

[bashirk]

Description

This PR adds inline documentation to the trainer.py file.

Relevant Issue

Issue #269

Checklist

• [x] I have followed the Contribution Guidelines and Code of Conduct
• [x] I have commented my code following the NimbleEdge Styleguide

[netlify[bot]]

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

Name	Link
🔨 Latest commit	6acb1cd727923dbaf0ed8683ba2988f4e3de505c
🔍 Latest deploy log	https://app.netlify.com/sites/goofy-agnesi-4565cb/deploys/62e3f2ced5d25d000889357d

[divyanshugit]

LGTM!! @ramesht007