admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 09:46:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: rename losses

Browse Source
This commit is contained in:
Khalil Meftah
2026-03-19 12:36:02 +01:00
parent d3e6f14d4f
commit aaf8576411
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rl/algorithms/sac/sac_algorithm.py
+4 -4
View File
@@ -189,7 +189,7 @@ class SACAlgorithm(RLAlgorithm):
critic_loss_val = loss_critic.item() critic_loss_val = loss_critic.item()
stats = TrainingStats( stats = TrainingStats(
losses={"critic": critic_loss_val}, losses={"loss_critic": critic_loss_val},
grad_norms={"critic": critic_grad_norm}, grad_norms={"critic": critic_grad_norm},
) )
@@ -202,7 +202,7 @@ class SACAlgorithm(RLAlgorithm):
max_norm=self.config.clip_grad_norm, max_norm=self.config.clip_grad_norm,
).item() ).item()
self.optimizers["discrete_critic"].step() self.optimizers["discrete_critic"].step()
stats.losses["discrete_critic"] = loss_discrete.item() stats.losses["loss_discrete_critic"] = loss_discrete.item()
stats.grad_norms["discrete_critic"] = dc_grad stats.grad_norms["discrete_critic"] = dc_grad
if self._optimization_step % self.config.policy_update_freq == 0: if self._optimization_step % self.config.policy_update_freq == 0:
@@ -225,8 +225,8 @@ class SACAlgorithm(RLAlgorithm):
).item() ).item()
self.optimizers["temperature"].step() self.optimizers["temperature"].step()
stats.losses["actor"] = actor_loss.item() stats.losses["loss_actor"] = actor_loss.item()
stats.losses["temperature"] = temp_loss.item() stats.losses["loss_temperature"] = temp_loss.item()
stats.grad_norms["actor"] = actor_grad stats.grad_norms["actor"] = actor_grad
stats.grad_norms["temperature"] = temp_grad stats.grad_norms["temperature"] = temp_grad
stats.extra["temperature"] = self.temperature stats.extra["temperature"] = self.temperature