[Cleanup] MethodType in baselines files replaced in favor of class forward functions in methods

[sami-bg]

Description

Addresses PR comment in #394

Checklist

• I have read the Contributing document.
• The documentation is up-to-date with the changes I made (check build artifacts).
• All tests passed, and additional code has been covered with new tests.
• I have added the PR to the RELEASES.rst file.

[RandallBalestriero]

Hi @sami-bg ! Thanks for the PR, quick question, what is the motivation for the naming of forward_model and use of this method instead of the usual forward?

[sami-bg]

Hello hello! I was sticking with the convention of having the forward function be reserved for one that operates directly from the batch dict (eg sticking to (self, batch, stage) arg signature), although now that you mention it, it would be useful to merge forward_model into forward since forward_model is what computes the loss. Will address that soon!

[sami-bg]

Fixed!