xpag.agents.rljax_agents.util.optim.optimize#
- optimize(fn_loss, opt, opt_state, params_to_update, max_grad_norm, *args, **kwargs)#
- Return type:
Tuple[Any,Mapping[str,Mapping[str,Array]],Array,Any]
Back to top