added wandb logger

[fatwir]

What does this PR do?

there are no breaking changes
tested PR locally using the pytest command
Fixes #<issue_number>

Before submitting

• Did you make sure title is self-explanatory and the description concisely explains the PR?
• Did you make sure your PR does only one thing, instead of bundling different changes together?
• Did you list all the breaking changes introduced by this pull request?
• Did you test your PR locally with pytest command?
• Did you run pre-commit hooks with pre-commit run -a command?

Did you have fun?

Make sure you had fun coding 🙃

[fatwir]

To Do

• Modify the current process to include testing in my existing run.
• Currently, when training completes, the run finishes, and I need to initialize a separate run for testing (this has not been done yet but needs to be).
• Integrate the testing procedure so that it occurs immediately after training completes, all within the same run.