torchgfn
torchgfn copied to clipboard
Add helper methods `policy_parameters()` and `logz_parameters()` helper functions.
module.parameters()
will by default return both of these as a single list / dict, but it is common to give a unique learning rate to the logZ
parameter (for example).
So we should provide the user with policy_parameters()
and logz_parameters()
(at least).