diff --git a/benchmarl/algorithms/ippo.py b/benchmarl/algorithms/ippo.py index c21c4faa..881d3d91 100644 --- a/benchmarl/algorithms/ippo.py +++ b/benchmarl/algorithms/ippo.py @@ -234,8 +234,8 @@ def process_batch(self, group: str, batch: TensorDictBase) -> TensorDictBase: loss = self.get_loss_and_updater(group)[0] loss.value_estimator( batch, - params=loss.critic_params, - target_params=loss.target_critic_params, + params=loss.critic_network_params, + target_params=loss.target_critic_network_params, ) return batch diff --git a/benchmarl/algorithms/mappo.py b/benchmarl/algorithms/mappo.py index 3dbfbed4..7e9031e4 100644 --- a/benchmarl/algorithms/mappo.py +++ b/benchmarl/algorithms/mappo.py @@ -238,8 +238,8 @@ def process_batch(self, group: str, batch: TensorDictBase) -> TensorDictBase: loss = self.get_loss_and_updater(group)[0] loss.value_estimator( batch, - params=loss.critic_params, - target_params=loss.target_critic_params, + params=loss.critic_network_params, + target_params=loss.target_critic_network_params, ) return batch